From 4d223e18a2600ca2467fb21ef4c18f0e9aa0d04c Mon Sep 17 00:00:00 2001 From: oahzxl Date: Mon, 9 Jan 2023 13:46:17 +0800 Subject: [PATCH] fix typo --- colossalai/autochunk/{estiamte_memory.py => estimate_memory.py} | 0 colossalai/autochunk/search_chunk.py | 2 +- colossalai/autochunk/select_chunk.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename colossalai/autochunk/{estiamte_memory.py => estimate_memory.py} (100%) 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