Adjust test directives for HardSwish to expect failure for integer ty…

…pes on CPU devices
Lightning-AI · Mar 28, 2024 · 4109be3 · 4109be3
1 parent e659c35
commit 4109be3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py
@@ -1628,6 +1628,13 @@ def hardswish_error_generator(op, device, dtype=torch.float32, **kwargs):
             dtypes=(datatypes.float16,),
             devicetypes=(devices.DeviceType.CPU,),
         ),
+        # PyTorch does not support CPU integer types hardswish
+        DecorateInfo(
+            pytest.mark.xfail,
+            "test_core_vs_torch_consistency",
+            dtypes=(datatypes.int16, datatypes.int32, datatypes.int64, datatypes.int8, datatypes.uint8),
+            devicetypes=(devices.DeviceType.CPU,),
+        ),
         # TODO: we might have a tolerance issue here with hardsiwsh, a function of relu6
         DecorateInfo(
             pytest.mark.xfail(strict=False),