diff --git a/colossalai/autochunk/estiamte_memory.py b/colossalai/autochunk/estimate_memory.py similarity index 100% rename from colossalai/autochunk/estiamte_memory.py rename to colossalai/autochunk/estimate_memory.py diff --git a/colossalai/autochunk/search_chunk.py b/colossalai/autochunk/search_chunk.py index e2c8de74e..21b967497 100644 --- a/colossalai/autochunk/search_chunk.py +++ b/colossalai/autochunk/search_chunk.py @@ -1,6 +1,6 @@ import copy -from .estiamte_memory import EstimateMemory +from .estimate_memory import EstimateMemory from .reorder_graph import ReorderGraph from .select_chunk import SelectChunk from .trace_flow import TraceFlow diff --git a/colossalai/autochunk/select_chunk.py b/colossalai/autochunk/select_chunk.py index bdc64528e..7127cfd64 100644 --- a/colossalai/autochunk/select_chunk.py +++ b/colossalai/autochunk/select_chunk.py @@ -1,4 +1,4 @@ -from .estiamte_memory import EstimateMemory +from .estimate_memory import EstimateMemory from .reorder_graph import ReorderGraph from .trace_index import TraceIndex from .utils import is_non_compute_node