[hotfix] Fix wrong import in meta_registry (#5392)

hpcaitech · Feb 20, 2024 · 5d380a1 · 5d380a1
1 parent b833153
commit 5d380a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/colossalai/auto_parallel/meta_profiler/meta_registry/binary_elementwise_ops.py b/colossalai/auto_parallel/meta_profiler/meta_registry/binary_elementwise_ops.py
@@ -4,9 +4,9 @@
 
 from colossalai._analyzer._subclasses.flop_tensor import flop_mapping
 from colossalai._analyzer.fx.node_util import compute_size_in_bytes as activation_size
+from colossalai.auto_parallel.tensor_shard.constants import BCAST_FUNC_OP
 from colossalai.auto_parallel.tensor_shard.sharding_strategy import MemoryCost, OperationDataType, TrainCycleItem
 
-from ..constants import BCAST_FUNC_OP
 from ..registry import meta_register
 
 __all__ = ["binary_elementwise_meta_info"]