Fix typo in native_batch_norm (#11446)

manuelcandales · web-flow · commit 42195dadb6a3 · 2025-06-25T17:11:00.000-04:00
Differential Revision: D76073674
diff --git a/kernels/portable/cpu/util/normalization_ops_util.cpp b/kernels/portable/cpu/util/normalization_ops_util.cpp
@@ -38,7 +38,7 @@ bool check_batch_norm_args(
     ET_LOG_AND_RETURN_IF_FALSE(
         tensors_have_same_dtype(in, running_mean.value()));
   }
-  if (running_mean.has_value()) {
+  if (running_var.has_value()) {
     ET_LOG_AND_RETURN_IF_FALSE(
         tensors_have_same_dtype(in, running_var.value()));
   }

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ bool check_batch_norm_args(`
`38`	`38`	`ET_LOG_AND_RETURN_IF_FALSE(`
`39`	`39`	`tensors_have_same_dtype(in, running_mean.value()));`
`40`	`40`	`}`
`41`		`- if (running_mean.has_value()) {`
	`41`	`+ if (running_var.has_value()) {`
`42`	`42`	`ET_LOG_AND_RETURN_IF_FALSE(`
`43`	`43`	`tensors_have_same_dtype(in, running_var.value()));`
`44`	`44`	`}`