Commit164acfa

authored

[None][infra] Skip test_eagle3 test with device memory check (NVIDIA#6617)

Signed-off-by: leslie-fang25 <leslief@nvidia.com>

1 parent7625845 commit164acfaCopy full SHA for 164acfa

File tree

-0

lines changed

-0

lines changed

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -1955,6 +1955,7 @@ def test_bf16(self, tp_size, pp_size, ep_size, attention_dp, cuda_graph,`
`1955`	`1955`	`task=MMLU(self.MODEL_NAME)`
`1956`	`1956`	`task.evaluate(llm)`
`1957`	`1957`
	`1958`	`+@pytest.mark.skip_less_device_memory(140000)## OOM on 80G H100`
`1958`	`1959`	`@parametrize_with_ids("eagle3_one_model", [True,False])`
`1959`	`1960`	`@parametrize_with_ids("enable_chunked_prefill", [False,True])`
`1960`	`1961`	`deftest_eagle3(self,enable_chunked_prefill,eagle3_one_model):`

Comments

(0)