[easy] Fix endif comments in functional_base.h (#153696)

swolchok · pytorchmergebot · commit 9ccd601a1486 · 2025-05-16T14:08:41.000Z
The first one of these confused me on #152388. Happened to notice the second. Pull Request resolved: #153696 Approved by: https://github.com/Skylion007, https://github.com/malfet
diff --git a/aten/src/ATen/cpu/vec/functional_base.h b/aten/src/ATen/cpu/vec/functional_base.h
@@ -140,7 +140,8 @@ struct VecReduceAllSIMD<float, std::plus<Vectorized<float>>> {
     return vaddvq_f32(acc_vec);
   }
 };
-#endif // defined(__aarch64__)
+#endif // defined(__aarch64__) && !defined(C10_MOBILE) && !defined(__CUDACC__)
+       // && !defined(CPU_CAPABILITY_SVE)
 
 #if defined(__aarch64__) && !defined(C10_MOBILE) && !defined(__CUDACC__) && \
     defined(CPU_CAPABILITY_SVE256)
@@ -166,7 +167,8 @@ struct VecReduceAllSIMD<float, Op> {
     return svlasta(svpfalse(), v);
   }
 };
-#endif // defined(__aarch64__)
+#endif // defined(__aarch64__) && !defined(C10_MOBILE) && !defined(__CUDACC__)
+       // && defined(CPU_CAPABILITY_SVE256)
 
 template <typename scalar_t, typename Op>
 inline scalar_t vec_reduce_all(
