We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 31c3c4c commit f9f2d97Copy full SHA for f9f2d97
setup.py
@@ -315,15 +315,13 @@ def get_extensions():
315
hip_source_dirs = [
316
os.path.join(extensions_dir, "cuda", "tensor_core_tiled_layout"),
317
# TODO: Add sparse_marlin back in once we have a ROCm build for it
318
- #os.path.join(extensions_dir, "cuda", "sparse_marlin")
+ # os.path.join(extensions_dir, "cuda", "sparse_marlin")
319
]
320
-
+
321
# Collect all HIP sources from the defined directories
322
hip_sources = []
323
for hip_dir in hip_source_dirs:
324
- hip_sources.extend(
325
- glob.glob(os.path.join(hip_dir, "*.cu"), recursive=True)
326
- )
+ hip_sources.extend(glob.glob(os.path.join(hip_dir, "*.cu"), recursive=True))
327
328
# Collect CUDA source files if needed
329
if not IS_ROCM and use_cuda:
0 commit comments