Fix a potential error in group norm (#2721)

intel · Jun 7, 2024 · 1885741 · 1885741
1 parent ccf4fb6
commit 1885741
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/itex/python/ops/group_norm_k3.py b/itex/python/ops/group_norm_k3.py
@@ -119,6 +119,7 @@ def __init__(
         self.use_gpu = None
 
     def build(self, input_shape):
+        self.supports_jit = False
         self.use_gpu = len(tf.config.list_physical_devices("XPU")) > 0
         dim = input_shape[self.axis]