From 25204fa7c5810381a0b5534bff1b13477ca6cb69 Mon Sep 17 00:00:00 2001
From: Wuwei Lin <wuwei@apache.org>
Date: Wed, 17 Jul 2024 16:18:46 -0700
Subject: [PATCH] fix

---
 python/tvm/relax/transform/__init__.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/python/tvm/relax/transform/__init__.py b/python/tvm/relax/transform/__init__.py
index eef6d331375c4..05e1376fb6972 100644
--- 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