Skip to content

Commit 26d36ec

Browse files
ZhanruiSunChmikeiovine
authored andcommitted
[None][infra] Remove invaild waived tests which not in release branch (NVIDIA#8841)
Signed-off-by: ZhanruiSunCh <[email protected]> Signed-off-by: Mike Iovine <[email protected]>
1 parent 0f5b086 commit 26d36ec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unittest/_torch/thop/parallel/test_fp8_rowwise_linear.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
import pytest
22
import torch
3-
from utils.util import skip_pre_hopper
3+
from utils.util import skip_blackwell
44

55
from tensorrt_llm._torch.modules.linear import Linear
66
from tensorrt_llm.models.modeling_utils import QuantAlgo, QuantConfig
77

88

9-
@skip_pre_hopper
9+
@skip_blackwell
1010
@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16])
1111
def test_fp8_rowwise_linear(dtype):
1212
SEQ_LEN = 10

0 commit comments

Comments
 (0)