Skip to content

Commit b626186

Browse files
authored
tests: fix some typo and limitation on test cases (#4989)
Signed-off-by: Ivy Zhang <[email protected]>
1 parent 9f5b23a commit b626186

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

tests/integration/defs/examples/test_gemma.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,7 @@ def test_llm_hf_gemma_quantization_1gpu_vswa(batch_size, data_type,
9191
gemma_example_root,
9292
llm_datasets_root, llm_rouge_root,
9393
qformat):
94+
skip_fp8_pre_ada(use_fp8=qformat == "fp8")
9495
max_attention_window = VSWA_ATTENTION[Path(gemma_model_root).stem]
9596
hf_gemma_quantization_1gpu(batch_size, data_type, gemma_model_root,
9697
llm_venv, cmodel_dir, engine_dir,

tests/integration/defs/test_e2e.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -829,7 +829,7 @@ def test_trtllm_bench_mgmn(llm_root, llm_venv):
829829
delete_on_close=True) as running_log:
830830
check_call(benchmark_cmd,
831831
shell=True,
832-
running_log=running_log,
832+
stdout=running_log,
833833
env=llm_venv._new_env)
834834
_check_mem_usage(running_log, [30, 0, 0, 0])
835835

0 commit comments

Comments
 (0)