File tree Expand file tree Collapse file tree 1 file changed +18
-2
lines changed
projects/rocprofiler-compute/src/utils Expand file tree Collapse file tree 1 file changed +18
-2
lines changed Original file line number Diff line number Diff line change 7373
7474
7575def 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
86102def detect_gpu_chip_id (rocminfo_lines : list [str ]) -> Optional [str ]:
You can’t perform that action at this time.
0 commit comments