fix

apache · Jul 17, 2024 · 25204fa · 25204fa
1 parent 5d684b8
commit 25204fa
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/python/tvm/relax/transform/__init__.py b/python/tvm/relax/transform/__init__.py
@@ -16,15 +16,6 @@
 # under the License.
 """Relax transformations. """
 
-# Import to register the legalization functions.
-from . import legalize_ops, tuning_api
-from .attach_external_modules import AttachExternModules
-from .fast_math import FastMathTransform
-from .ipc_allreduce_rewrite import IPCAllReduceRewrite
-from .lazy_transform_params import LazyTransformParams
-from .lower_gpu_ipc_alloc_storage import LowerGPUIPCAllocStorage
-from .optimize_layout_transform import OptimizeLayoutTransform
-from .remove_redundant_reshape import RemoveRedundantReshape
 from .transform import (
     AdjustMatmulOrder,
     AllocateWorkspace,
@@ -92,3 +83,14 @@
     dataflowblock_pass,
     function_pass,
 )
+
+from .attach_external_modules import AttachExternModules
+from .fast_math import FastMathTransform
+from .ipc_allreduce_rewrite import IPCAllReduceRewrite
+from .lazy_transform_params import LazyTransformParams
+from .lower_gpu_ipc_alloc_storage import LowerGPUIPCAllocStorage
+from .optimize_layout_transform import OptimizeLayoutTransform
+from .remove_redundant_reshape import RemoveRedundantReshape
+
+# Import to register the legalization functions.
+from . import legalize_ops, tuning_api