File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments