Skip to content

Commit 5da42c9

Browse files
committed
mx inference test: fix on B200
Summary: Skip a test which is failing locally on a B200 machine Test Plan: ``` pytest test/prototype/mx_formats -s -x ``` Reviewers: Subscribers: Tasks: Tags: ghstack-source-id: c0e47ba ghstack-comment-id: 3348186698 Pull-Request: #3093
1 parent 75944ab commit 5da42c9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/prototype/mx_formats/test_inference_workflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def test_inference_workflow_mx(elem_dtype, bias: bool, compile: bool, emulate: b
7070
elif elem_dtype == torch.float4_e2m1fn_x2:
7171
if not is_sm_at_least_100() and not emulate:
7272
pytest.skip("CUDA capability >= 10.0 required for mxfp4 gemm")
73-
elif not is_sm_at_least_100() and emulate and compile:
73+
elif emulate and compile:
7474
# TODO(future PR): investigate and fix this
7575
pytest.skip("mxfp4 + emulate + compile currently does not work, low SQNR")
7676

0 commit comments

Comments
 (0)