Skip to content

Commit f9f2d97

Browse files
committed
Refactor HIP source collection in setup.py for improved readability
1 parent 31c3c4c commit f9f2d97

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

setup.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -315,15 +315,13 @@ def get_extensions():
315315
hip_source_dirs = [
316316
os.path.join(extensions_dir, "cuda", "tensor_core_tiled_layout"),
317317
# TODO: Add sparse_marlin back in once we have a ROCm build for it
318-
#os.path.join(extensions_dir, "cuda", "sparse_marlin")
318+
# os.path.join(extensions_dir, "cuda", "sparse_marlin")
319319
]
320-
320+
321321
# Collect all HIP sources from the defined directories
322322
hip_sources = []
323323
for hip_dir in hip_source_dirs:
324-
hip_sources.extend(
325-
glob.glob(os.path.join(hip_dir, "*.cu"), recursive=True)
326-
)
324+
hip_sources.extend(glob.glob(os.path.join(hip_dir, "*.cu"), recursive=True))
327325

328326
# Collect CUDA source files if needed
329327
if not IS_ROCM and use_cuda:

0 commit comments

Comments
 (0)