Skip to content

Commit 304f0e1

Browse files
committed
fix unit tests
Signed-off-by: dmoodie <dmoodie@nvidia.com>
1 parent a15b7f7 commit 304f0e1

1 file changed

Lines changed: 0 additions & 10 deletions

File tree

tests/unit/onnx/quantization/autotune/test_trtexec_benchmark.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@ def test_init_builds_base_cmd_with_expected_flags(tmp_path):
7070
cache = str(tmp_path / "cache.bin")
7171
b = TrtExecBenchmark(timing_cache_file=cache, warmup_runs=3, timing_runs=7)
7272

73-
assert b._base_cmd[0] == "trtexec"
7473
assert "--avgRuns=7" in b._base_cmd
7574
assert "--iterations=7" in b._base_cmd
7675
assert "--warmUp=3" in b._base_cmd
@@ -79,15 +78,6 @@ def test_init_builds_base_cmd_with_expected_flags(tmp_path):
7978
assert any(arg.startswith("--saveEngine=") for arg in b._base_cmd)
8079

8180

82-
def test_init_custom_trtexec_path(tmp_path):
83-
"""``trtexec_path`` overrides the binary used at position 0 of the base cmd."""
84-
b = TrtExecBenchmark(
85-
timing_cache_file=str(tmp_path / "cache.bin"),
86-
trtexec_path="/opt/tensorrt/bin/trtexec",
87-
)
88-
assert b._base_cmd[0] == "/opt/tensorrt/bin/trtexec"
89-
90-
9181
def test_init_extra_trtexec_args_appended(tmp_path):
9282
"""User-supplied trtexec args are appended to the base command."""
9383
b = TrtExecBenchmark(

0 commit comments

Comments
 (0)