File tree Expand file tree Collapse file tree 2 files changed +1
-3
lines changed
unittest/_torch/thop/parallel Expand file tree Collapse file tree 2 files changed +1
-3
lines changed Original file line number Diff line number Diff line change @@ -274,7 +274,6 @@ accuracy/test_cli_flow.py::TestPhi4MiniInstruct::test_tp2 SKIP (https://nvbugs/5
274274accuracy/test_cli_flow.py::TestLongAlpaca7B::test_auto_dtype SKIP (https://nvbugs/5481075)
275275accuracy/test_llm_api.py::TestPhi4MiniInstruct::test_fp8 SKIP (https://nvbugs/5465143, 5481206 WNF)
276276accuracy/test_llm_api_pytorch.py::TestEXAONE4::test_auto_dtype SKIP (https://nvbugs/5481090)
277- accuracy/test_disaggregated_serving.py::TestLlama3_1_8BInstruct::test_ngram SKIP (https://nvbugs/5488118)
278277test_e2e.py::test_trtllm_bench_iteration_log[TRT-streaming-meta-llama/Llama-3.1-8B-llama-3.1-model/Meta-Llama-3.1-8B] SKIP (https://nvbugs/5448523)
279278cpp/test_unit_tests.py::test_unit_tests[kernels-80] SKIP (https://nvbugs/5504078)
280279accuracy/test_llm_api_pytorch.py::TestLlama3_2_3B::test_auto_dtype SKIP (https://nvbugs/5520319)
Original file line number Diff line number Diff line change 11import pytest
22import torch
3- from utils .util import skip_pre_hopper
43
54from tensorrt_llm ._torch .modules .linear import Linear
65from tensorrt_llm .models .modeling_utils import QuantAlgo , QuantConfig
76
87
9- @skip_pre_hopper
8+ @skip_blackwell
109@pytest .mark .parametrize ("dtype" , [torch .float16 , torch .bfloat16 ])
1110def test_fp8_rowwise_linear (dtype ):
1211 SEQ_LEN = 10
You can’t perform that action at this time.
0 commit comments