Skip to content

Commit d4b2f33

Browse files
authored
Skip test_fpx_weight_only in fbcode (#1056)
1 parent db72dd1 commit d4b2f33

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/dtypes/test_floatx.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
fpx_weight_only,
2222
)
2323

24-
from torchao.utils import TORCH_VERSION_AT_LEAST_2_5
24+
from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, is_fbcode
2525

2626

2727
_DEVICES = ["cpu"] + (["cuda"] if torch.cuda.is_available() else [])
@@ -91,6 +91,7 @@ def test_to_copy_device(self, ebits, mbits):
9191
@pytest.mark.skipif(not TORCH_VERSION_AT_LEAST_2_5, reason="quantization only works with torch.compile for 2.5+")
9292
@parametrize("ebits,mbits", _Floatx_DTYPES)
9393
@parametrize("bias", [False, True])
94+
@pytest.mark.skipif(is_fbcode(), reason="broken in fbcode")
9495
def test_fpx_weight_only(self, ebits, mbits, bias):
9596
N, OC, IC = 4, 256, 64
9697
device = "cuda"

0 commit comments

Comments
 (0)