Remote OptimizeBatchnorm pass. Redundant with DecomposeOpsForInference.

apache · Jan 28, 2025 · 31819e6 · 31819e6
1 parent 89f0dd8
commit 31819e6
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 106 deletions.
diff --git a/python/tvm/relax/transform/__init__.py b/python/tvm/relax/transform/__init__.py
@@ -96,7 +96,6 @@
 from .lazy_transform_params import LazyTransformParams
 from .lower_gpu_ipc_alloc_storage import LowerGPUIPCAllocStorage
 from .optimize_layout_transform import OptimizeLayoutTransform
-from .optimize_batchnorm import OptimizeBatchnorm
 from .remove_redundant_reshape import RemoveRedundantReshape
 
 # Import to register the legalization functions.

diff --git a/python/tvm/relax/transform/optimize_batchnorm.py b/python/tvm/relax/transform/optimize_batchnorm.py
diff --git a/tests/python/relax/test_transform_annotate_custom_scope.py b/tests/python/relax/test_transform_annotate_custom_scope.py
@@ -69,8 +69,6 @@ def verify(mod, expected):
     tgt = tvm.target.Target("opencl --device=adreno", host="llvm")
     with tgt:
         mod = tvm.tir.transform.BindTarget(tvm.target.Target.current(allow_none=False))(mod)
-        mod = tvm.relax.transform.OptimizeBatchnorm()(mod)
-        mod = tvm.relax.transform.FoldConstant()(mod)
         mod = tvm.relax.transform.DecomposeOpsForInference()(mod)
         mod = tvm.relax.transform.FoldConstant()(mod)
         desired_layouts = {"relax.nn.conv2d": ["NCHW4c", "OIHW4o", "NCHW4c"]}