From 0b6448395a62ecc1cf02e482964e27426558e155 Mon Sep 17 00:00:00 2001
From: Utkarsh <49331882+uppalutkarsh@users.noreply.github.com>
Date: Thu, 28 Dec 2023 16:46:16 +0530
Subject: [PATCH] Add custom gRPC url and port argument

---
 src/c++/perf_analyzer/docs/examples/profile.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/c++/perf_analyzer/docs/examples/profile.py b/src/c++/perf_analyzer/docs/examples/profile.py
index 7b79c9848..695ec5e5d 100644
--- 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)