Skip to content

Commit 9efd330

Browse files
authored
add warning msg for unsupported arch in profile mode. (#1933)
1 parent 09010eb commit 9efd330

File tree

1 file changed

+18
-2
lines changed
  • projects/rocprofiler-compute/src/utils

1 file changed

+18
-2
lines changed

projects/rocprofiler-compute/src/utils/specs.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,14 +73,30 @@
7373

7474

7575
def detect_arch(rocminfo_lines: list[str]) -> Optional[tuple[str, int]]:
76+
supported_gpu_arch = mi_gpu_specs.get_gpu_series_dict()
77+
unsupported_gpu_arch: set[str] = set()
78+
7679
for idx1, line_text in enumerate(rocminfo_lines):
7780
gpu_arch = search(
7881
r"^\s*Name\s*:\s* ([Gg][Ff][Xx][a-zA-Z0-9]+).*\s*$", line_text
7982
)
80-
if gpu_arch and gpu_arch in mi_gpu_specs.get_gpu_series_dict():
83+
if not gpu_arch:
84+
continue
85+
86+
if gpu_arch in supported_gpu_arch:
8187
return (gpu_arch, idx1)
8288

83-
console_error("Cannot find a supported arch in rocminfo")
89+
if gpu_arch not in unsupported_gpu_arch:
90+
unsupported_gpu_arch.add(gpu_arch)
91+
console_warning(
92+
"Detected GPU architecture: "
93+
f"{gpu_arch} is currently NOT supported by the profile mode."
94+
)
95+
96+
if unsupported_gpu_arch:
97+
console_log(f"Supported architectures: {list(supported_gpu_arch.keys())}")
98+
99+
console_error("Cannot find a supported arch in rocminfo.")
84100

85101

86102
def detect_gpu_chip_id(rocminfo_lines: list[str]) -> Optional[str]:

0 commit comments

Comments
 (0)