Update headers inclusion as per review comments

pytorch · Feb 24, 2025 · 59511cd · 59511cd
1 parent a8829e2
commit 59511cd
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/torchao/experimental/ops/parallel-aten-impl.h b/torchao/experimental/ops/parallel-aten-impl.h
@@ -5,11 +5,7 @@
 // LICENSE file in the root directory of this source tree.
 
 #pragma once
-#if defined(__aarch64__)
-    #include <torchao/experimental/ops/parallel.h>
-#else
-    #include <Aten/Parallel.h>
-#endif
+#include <torchao/experimental/ops/parallel.h>
 #include <torch/library.h>
 #include <torch/torch.h>