Skip to content

Commit e1a100c

Browse files
committed
fix test
Signed-off-by: Dylan Chen <[email protected]>
1 parent 711d3d9 commit e1a100c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unittest/_torch/thop/test_fp8_rowwise_linear.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22
import torch
3-
from utils.util import skip_pre_hopper
3+
from utils.util import skip_blackwell, skip_pre_hopper
44

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

0 commit comments

Comments
 (0)