Commit 21a6844
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.dat1 parent a9f6655 commit 21a6844
1 file changed
+72
-543
lines changed
0 commit comments