Remove Gelu Fusion for TF Newapi (#1886)

Signed-off-by: zehao-intel <[email protected]>
intel · Jul 2, 2024 · 5592acc · 5592acc
1 parent 4372a76
commit 5592acc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 415 deletions.
diff --git a/neural_compressor/tensorflow/quantization/utils/graph_rewriter/generic/fuse_gelu.py b/neural_compressor/tensorflow/quantization/utils/graph_rewriter/generic/fuse_gelu.py
@@ -22,15 +22,14 @@
 from neural_compressor.tensorflow.quantization.utils.graph_rewriter.graph_base import GraphRewriterBase
 from neural_compressor.tensorflow.quantization.utils.graph_util import GraphAnalyzer
 from neural_compressor.tensorflow.quantization.utils.graph_util import GraphRewriterHelper as Helper
-from neural_compressor.tensorflow.utils import SPR_BASE_VERSIONS
 
 
 class FuseGeluOptimizer(GraphRewriterBase):  # pragma: no cover
     """Fuse Sqrt + RealDiv + Erf + AddV2 + Mul + Mul into Gelu op."""
 
     def do_transformation(self):
         """Execute the fusion from small ops to Gelu."""
-        if not (tf.version.VERSION in ("1.15.0-up2", "1.15.0-up3") or tf.version.VERSION in SPR_BASE_VERSIONS):
+        if tf.version.VERSION not in ("1.15.0-up2", "1.15.0-up3"):
             return self.model
 
         cur_graph = GraphAnalyzer()