Fix

co63oc · co63oc · commit 3ae95d041efb · 2025-07-19T14:18:43.000+08:00
diff --git a/paddle/phi/kernels/cpu/batch_norm_kernel.cc b/paddle/phi/kernels/cpu/batch_norm_kernel.cc
@@ -58,6 +58,7 @@ void BatchNormKernel(const Context& dev_ctx,
     if (saved_mean) dev_ctx.template Alloc<T>(saved_mean);
     if (saved_variance) dev_ctx.template Alloc<T>(saved_variance);
     if (reserve_space) {
+      // infermeta dim is -1.
       reserve_space->Resize({0});
       dev_ctx.template Alloc<T>(reserve_space);
     }

Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,7 @@ void BatchNormKernel(const Context& dev_ctx,`
`58`	`58`	`if (saved_mean) dev_ctx.template Alloc<T>(saved_mean);`
`59`	`59`	`if (saved_variance) dev_ctx.template Alloc<T>(saved_variance);`
`60`	`60`	`if (reserve_space) {`
	`61`	`+ // infermeta dim is -1.`
`61`	`62`	`reserve_space->Resize({0});`
`62`	`63`	`dev_ctx.template Alloc<T>(reserve_space);`
`63`	`64`	`}`