Skip to content

Commit 6a94c2b

Browse files
committed
Fix LLM profiling
Update copyright
1 parent b44135a commit 6a94c2b

File tree

1 file changed

+4
-16
lines changed

1 file changed

+4
-16
lines changed

model_analyzer/perf_analyzer/perf_analyzer.py

Lines changed: 4 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,6 @@
11
#!/usr/bin/env python3
2-
3-
# Copyright 2020-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
4-
#
5-
# Licensed under the Apache License, Version 2.0 (the "License");
6-
# you may not use this file except in compliance with the License.
7-
# You may obtain a copy of the License at
8-
#
9-
# http://www.apache.org/licenses/LICENSE-2.0
10-
#
11-
# Unless required by applicable law or agreed to in writing, software
12-
# distributed under the License is distributed on an "AS IS" BASIS,
13-
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14-
# See the License for the specific language governing permissions and
15-
# limitations under the License.
2+
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3+
# SPDX-License-Identifier: Apache-2.0
164

175
import csv
186
import glob
@@ -432,8 +420,8 @@ def _get_cmd(self):
432420
return cmd
433421

434422
def _get_single_model_cmd(self, index):
435-
if self._model_type == "LLM":
436-
cmd = ["genai-perf", "-m", self._config.models_name()]
423+
if self._model_type.lower() == "llm":
424+
cmd = ["genai-perf", "profile", "-m", self._config.models_name()]
437425
cmd += self._get_genai_perf_cli_command(index).replace("=", " ").split()
438426
cmd += ["--"]
439427
cmd += (

0 commit comments

Comments
 (0)