From 9116a19950bff249dcf5c96cc2e5315bd692a063 Mon Sep 17 00:00:00 2001
From: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Date: Wed, 19 Feb 2025 15:59:34 +0000
Subject: [PATCH] Remove from custom kernels as well

---
 csrc/rocm/custom_kernels.cu | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/csrc/rocm/custom_kernels.cu b/csrc/rocm/custom_kernels.cu
index 6143729b237f4..3533108b3316a 100644
--- 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