File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -308,9 +308,10 @@ def compile_and_sort_results(self):
308308 ]
309309 for model_name in analysis_model_names :
310310 if model_name not in results :
311- logger . warning (
311+ raise TritonModelAnalyzerException (
312312 f"Model { model_name } requested for analysis but no results were found. "
313- "Ensure that this model was actually profiled." )
313+ "Double check the name and ensure that this model was actually profiled."
314+ )
314315 else :
315316 result_dict = results [model_name ]
316317 for (model_config , measurements ) in result_dict .values ():
@@ -351,9 +352,10 @@ def get_model_config_measurements(self, model_config_name):
351352
352353 if model_name not in results or model_config_name not in results [
353354 model_name ]:
354- logger .error (
355- f'No results found for model config: { model_config_name } ' )
356- return (None , [])
355+ raise TritonModelAnalyzerException (
356+ f"Model Config { model_config_name } requested for report step but no results were found. "
357+ "Double check the name and ensure that this model config was actually profiled."
358+ )
357359 else :
358360 model_config_data = results [model_name ][model_config_name ]
359361 return model_config_data [0 ], list (model_config_data [1 ].values ())
You can’t perform that action at this time.
0 commit comments