[BUILD] Fix unsupported arguments in generator (#1519)

* Fix unsupported arguments in generator

* Get ROCM_PATH as env variable

[ROCm/rccl commit: 5804603632]
This commit is contained in:
Bertan Dogancay
2025-02-03 14:51:55 -05:00
committed by GitHub
parent 37409368f9
commit e171f59719
2 changed files with 5 additions and 2 deletions
+3 -1
View File
@@ -137,7 +137,9 @@ def calc_unroll_for_local_arch():
if not is_local_arch_only:
return
res = subprocess.run(['rocminfo'], capture_output=True, text=True)
rocminfo_path = os.environ.get('ROCM_PATH') + "/bin/rocminfo"
res = subprocess.run([rocminfo_path], stdout=subprocess.PIPE, universal_newlines=True)
rocminfo_output = res.stdout
# Parse rocminfo binary output