Skip to content

Commit e35ebb0

Browse files
committed
unit test
Signed-off-by: Fridah-nv <201670829+Fridah-nv@users.noreply.github.com>
1 parent eac8417 commit e35ebb0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/gpu/torch/quantization/test_quantize_cuda.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,7 @@
8787
mtq.NVFP4_AWQ_LITE_CFG,
8888
mtq.NVFP4_AWQ_CLIP_CFG,
8989
mtq.NVFP4_AWQ_FULL_CFG,
90+
mtq.NVFP4_LOCAL_HESSIAN_WEIGHT_ONLY_CFG,
9091
mtq.MXFP8_DEFAULT_CFG,
9192
mtq.MXFP6_DEFAULT_CFG,
9293
mtq.MXFP4_DEFAULT_CFG,
@@ -113,6 +114,7 @@ def test_quantize(model_cls, config):
113114
mtq.FP8_2D_BLOCKWISE_WEIGHT_ONLY_CFG,
114115
NVFP4_WEIGHT_ACT_MSE_CFG,
115116
NVFP4_WEIGHT_MSE_FP8_SWEEP_CFG,
117+
mtq.NVFP4_LOCAL_HESSIAN_WEIGHT_ONLY_CFG,
116118
]:
117119
if get_cuda_ext_mx() is None:
118120
pytest.skip("cuda_ext_mx is not available")

0 commit comments

Comments
 (0)