Remove from custom kernels as well

ROCm · Feb 19, 2025 · 9116a19 · 9116a19
1 parent a6d2b59
commit 9116a19
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/csrc/rocm/custom_kernels.cu b/csrc/rocm/custom_kernels.cu
@@ -5,13 +5,11 @@
 #include <algorithm>
 #include "cuda_compat.h"
 
-#if defined(__HIPCC__) && (defined(__gfx90a__) || defined(__gfx940__) || \
-                           defined(__gfx941__) || defined(__gfx942__))
+#if defined(__HIPCC__) && (defined(__gfx90a__) || defined(__gfx942__))
   #define __HIP__MI300_MI250__
 #endif
 
-#if defined(__HIPCC__) && \
-    (defined(__gfx940__) || defined(__gfx941__) || defined(__gfx942__))
+#if defined(__HIPCC__) && defined(__gfx942__)
   #define __HIP__MI300__
 #endif