Update

swolchok · swolchok · commit 3d626273769f · 2025-02-11T15:13:40.000-08:00
[ghstack-poisoned]
diff --git a/codegen/macros.h b/codegen/macros.h
@@ -8,6 +8,6 @@
 
 #pragma once
 // TODO(T157709949) remove this file
-#ifndef USE_ATEN_LIB
+#if !defined(USE_ATEN_LIB) && !defined(TORCH_API)
 #define TORCH_API
 #endif
