[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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user