Refactor HIP source directory handling in setup.py and remove deprecated runner configuration from build workflow

Author: petrex

.github/workflows/build_wheels_linux.yml

@@ -32,7 +32,6 @@ jobs:
       with-xpu: enable
       # Note: if free-threaded python is required add py3.13t here
       python-versions: '["3.9"]'
-      runner: '{"rocm": "linux.rocm.gpu.mi300.2"}'
 
   build:
     needs: generate-matrix

setup.py

@@ -311,16 +311,19 @@ def get_extensions():
         glob.glob(os.path.join(extensions_cuda_dir, "**/*.cu"), recursive=True)
     )
 
-    extensions_hip_dir = os.path.join(
-        extensions_dir, "cuda", "tensor_core_tiled_layout"
-    )
-    hip_sources = list(
-        glob.glob(os.path.join(extensions_hip_dir, "*.cu"), recursive=True)
-    )
-    extensions_hip_dir = os.path.join(extensions_dir, "cuda", "sparse_marlin")
-    hip_sources += list(
-        glob.glob(os.path.join(extensions_hip_dir, "*.cu"), recursive=True)
-    )
+    # Define HIP source directories
+    hip_source_dirs = [
+        os.path.join(extensions_dir, "cuda", "tensor_core_tiled_layout"),
+        # TODO: Add sparse_marlin back in once we have a ROCm build for it
+        #os.path.join(extensions_dir, "cuda", "sparse_marlin")
+    ]
+    
+    # Collect all HIP sources from the defined directories
+    hip_sources = []
+    for hip_dir in hip_source_dirs:
+        hip_sources.extend(
+            glob.glob(os.path.join(hip_dir, "*.cu"), recursive=True)
+        )
 
     # Collect CUDA source files if needed
     if not IS_ROCM and use_cuda: