Skip to content

Commit 3d447b6

Browse files
authored
Fix issue where report step cant find remote mode data (#339)
* Fix issue where report step cant find remote mode data * Fix bad options in test
1 parent 6077f4b commit 3d447b6

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

model_analyzer/result/result_manager.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -350,6 +350,11 @@ def get_model_config_measurements(self, model_config_name):
350350
#
351351
model_name = model_config_name.rsplit('_', 2)[0]
352352

353+
# Remote mode has model_name == model_config_name
354+
#
355+
if model_name not in results:
356+
model_name = model_config_name
357+
353358
if model_name not in results or model_config_name not in results[
354359
model_name]:
355360
raise TritonModelAnalyzerException(

qa/L0_config_range/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ MODEL_ANALYZER_PROFILE_BASE_ARGS="$MODEL_ANALYZER_PROFILE_BASE_ARGS --triton-htt
4141
MODEL_ANALYZER_PROFILE_BASE_ARGS="$MODEL_ANALYZER_PROFILE_BASE_ARGS --triton-metrics-url http://localhost:${PORTS[2]}/metrics --checkpoint-directory=$CHECKPOINT_DIRECTORY"
4242
MODEL_ANALYZER_PROFILE_BASE_ARGS="$MODEL_ANALYZER_PROFILE_BASE_ARGS --output-model-repository-path $OUTPUT_MODEL_REPOSITORY --override-output-model-repository"
4343

44-
MODEL_ANALYZER_ANALYZE_BASE_ARGS="--analysis-models $MODEL_NAMES -e $EXPORT_PATH --filename-server-only=$FILENAME_SERVER_ONLY --checkpoint-directory=$CHECKPOINT_DIRECTORY"
44+
MODEL_ANALYZER_ANALYZE_BASE_ARGS="-e $EXPORT_PATH --filename-server-only=$FILENAME_SERVER_ONLY --checkpoint-directory=$CHECKPOINT_DIRECTORY"
4545
MODEL_ANALYZER_ANALYZE_BASE_ARGS="$MODEL_ANALYZER_ANALYZE_BASE_ARGS --filename-model-inference=$FILENAME_INFERENCE_MODEL --filename-model-gpu=$FILENAME_GPU_MODEL"
4646

4747
python3 config_generator.py

0 commit comments

Comments
 (0)