Skip to content

Commit 21a6844

Browse files
committed
Refactor to remove switch statement in build_rocm.sh
This commit refactors build_rocm.sh and brings in a few major updates: - No longer required to specify the full .so name (with number) for ROCm libraries - The .so versions are copied and the patching code will fix the links to point to this version - No longer required to specify paths for ROCm libraries allowing the removal of the large switch - Paths are acquired programmatically with find - No longer required to specify both the path and filename for the OS specific libraries - Programatically extract file name from the path - Automatically extract Tensile/Kernels files for the architectures specified in PYTORCH_ROCM_ARCH and any non-arch specific files e.g. TensileLibrary.dat
1 parent a9f6655 commit 21a6844

File tree

1 file changed

+72
-543
lines changed

1 file changed

+72
-543
lines changed

0 commit comments

Comments
 (0)