@@ -25,6 +25,8 @@ TRITON_LAUNCH_MODE=${TRITON_LAUNCH_MODE:="remote"}
2525CLIENT_PROTOCOL=" http"
2626PORTS=(` find_available_ports 2` )
2727HTTP_PORT=" 8000"
28+ GRPC_PORT=" ${PORTS[0]} "
29+ METRICS_PORT=" ${PORTS[1]} "
2830GPUS=(` get_all_gpus_uuids` )
2931OUTPUT_MODEL_REPOSITORY=${OUTPUT_MODEL_REPOSITORY:= `get_output_directory`}
3032WORKING_CONFIG_FILE=" working_config.yml"
@@ -64,7 +66,7 @@ cp client.crt client2.crt && sed -i "s/\b\(.\)/\u\1/g" client2.crt
6466
6567# For remote launch, set server args and start server
6668SERVER=` which tritonserver`
67- SERVER_ARGS=" --model-repository=$MODEL_REPOSITORY --model-control-mode explicit --http-port ${HTTP_PORT} --grpc-port ${PORTS[0] } --metrics-port ${PORTS[1] } "
69+ SERVER_ARGS=" --model-repository=$MODEL_REPOSITORY --model-control-mode explicit --http-port ${HTTP_PORT} --grpc-port ${GRPC_PORT } --metrics-port ${METRICS_PORT } "
6870SERVER_HTTP_PORT=${HTTP_PORT}
6971SERVER_LOG=" $LOGS_DIR /server.log"
7072
@@ -87,8 +89,8 @@ create_result_paths -test-name $TEST_NAME
8789
8890MODEL_ANALYZER_ARGS=" -m $MODEL_REPOSITORY -f $WORKING_CONFIG_FILE -e $EXPORT_PATH --checkpoint-directory $CHECKPOINT_DIRECTORY "
8991MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --client-protocol=$CLIENT_PROTOCOL --triton-launch-mode=$TRITON_LAUNCH_MODE "
90- MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-http-endpoint https://localhost:443 --triton-grpc-endpoint localhost:${PORTS[1] } "
91- MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-metrics-url https ://localhost:${PORTS[2] } /metrics"
92+ MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-http-endpoint https://localhost:443 --triton-grpc-endpoint localhost:${GRPC_PORT } "
93+ MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-metrics-url http ://localhost:${METRICS_PORT } /metrics"
9294MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --output-model-repository-path $OUTPUT_MODEL_REPOSITORY --override-output-model-repository"
9395MODEL_ANALYZER_SUBCOMMAND=" profile"
9496run_analyzer
@@ -104,8 +106,8 @@ create_result_paths -test-name $TEST_NAME
104106
105107MODEL_ANALYZER_ARGS=" -m $MODEL_REPOSITORY -f $BROKEN_CONFIG_FILE -e $EXPORT_PATH --checkpoint-directory $CHECKPOINT_DIRECTORY "
106108MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --client-protocol=$CLIENT_PROTOCOL --triton-launch-mode=$TRITON_LAUNCH_MODE "
107- MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-http-endpoint https://localhost:443 --triton-grpc-endpoint localhost:${PORTS[1] } "
108- MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-metrics-url https ://localhost:${PORTS[2] } /metrics"
109+ MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-http-endpoint https://localhost:443 --triton-grpc-endpoint localhost:${GRPC_PORT } "
110+ MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --triton-metrics-url http ://localhost:${METRICS_PORT } /metrics"
109111MODEL_ANALYZER_ARGS=" $MODEL_ANALYZER_ARGS --output-model-repository-path $OUTPUT_MODEL_REPOSITORY --override-output-model-repository"
110112MODEL_ANALYZER_SUBCOMMAND=" profile"
111113run_analyzer
0 commit comments