Skip to content

Commit 9368b28

Browse files
authored
mx inference test: fix on B200 (#3093)
Update [ghstack-poisoned]
1 parent d407246 commit 9368b28

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)