|
1 | 1 | #!/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 |
16 | 4 |
|
17 | 5 | import csv |
18 | 6 | import glob |
@@ -432,8 +420,8 @@ def _get_cmd(self): |
432 | 420 | return cmd |
433 | 421 |
|
434 | 422 | 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()] |
437 | 425 | cmd += self._get_genai_perf_cli_command(index).replace("=", " ").split() |
438 | 426 | cmd += ["--"] |
439 | 427 | cmd += ( |
|
0 commit comments