From 4109be36712295be14f5fce0fa822691b6add239 Mon Sep 17 00:00:00 2001
From: Shahar <shaharelys@gmail.com>
Date: Fri, 29 Mar 2024 00:31:54 +0200
Subject: [PATCH] Adjust test directives for HardSwish to expect failure for
 integer types on CPU devices

---
 thunder/tests/opinfos.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py
index e9e92222ae..b7b615b3f6 100644
--- 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),