Fix typo in native_batch_norm (pytorch#11446)

manuelcandales · facebook-github-bot · commit e254036c58d9 · 2025-06-24T11:48:03.000-07:00
Summary: Pull Request resolved: pytorch#11446 Reviewed By: swolchok 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`	`}`