Skip to content

GitLab

Unverified Commit 5d380a1a authored by Stephan Kölker's avatar Stephan Kölker Committed by GitHub
Browse files

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

parent b833153f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/auto_parallel/meta_profiler/meta_registry/binary_elementwise_ops.py
View file @ 5d380a1a
...@@ -4,9 +4,9 @@ import torch ...@@ -4,9 +4,9 @@ import torch
from colossalai._analyzer._subclasses.flop_tensor import flop_mapping 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._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 colossalai.auto_parallel.tensor_shard.sharding_strategy import MemoryCost, OperationDataType, TrainCycleItem
from ..constants import BCAST_FUNC_OP
from ..registry import meta_register from ..registry import meta_register
__all__ = ["binary_elementwise_meta_info"] __all__ = ["binary_elementwise_meta_info"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment