Add custom gRPC url and port argument

triton-inference-server · Dec 28, 2023 · 0b64483 · 0b64483
1 parent b09488d
commit 0b64483
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/c++/perf_analyzer/docs/examples/profile.py b/src/c++/perf_analyzer/docs/examples/profile.py
@@ -417,7 +417,7 @@ def summarize_profile_results(args, prompts):
 
 def profile(args, export_file):
     command = (
-        f"perf_analyzer -m {args.model} -i grpc --async --streaming "
+        f"perf_analyzer -m {args.model} -u {args.url} -i grpc --async --streaming "
         f"--input-data={INPUT_FILENAME} "
         f"--profile-export-file={export_file} "
     )
@@ -645,5 +645,12 @@ def main(args):
         action="store_true",
         help="Whether to stop streaming the model outputs.",
     )
+    parser.add_argument(
+        "-u",
+        "--url",
+        type=str,
+        default="localhost:8001",
+        help="gRPC end point URL",
+    )
     args = parser.parse_args()
     main(args)