[BUILD] Fix unsupported arguments in generator (#1519)
* Fix unsupported arguments in generator * Get ROCM_PATH as env variable
Tento commit je obsažen v:
1 rodič
a5c6b547a2
revize
5804603632
2 files changed
+5
-2
No files matched your search
@@ -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
Zablokovat Uživatele