From 63f250bbd49adf5fac8f670bb98181f81e5d4369 Mon Sep 17 00:00:00 2001 From: Ziyue Jiang Date: Tue, 25 Oct 2022 16:48:48 +0800 Subject: [PATCH] fix file name (#1759) Co-authored-by: Ziyue Jiang --- colossalai/pipeline/__init__.py | 2 +- colossalai/pipeline/{layer_sepc.py => layer_spec.py} | 0 colossalai/pipeline/pipelinable.py | 2 +- docs/colossalai/colossalai.pipeline.layer_sepc.rst | 2 +- docs/colossalai/colossalai.pipeline.rst | 2 +- 5 files changed, 4 insertions(+), 4 deletions(-) rename colossalai/pipeline/{layer_sepc.py => layer_spec.py} (100%) diff --git a/colossalai/pipeline/__init__.py b/colossalai/pipeline/__init__.py index 625bd7ef5..0fcde9707 100644 --- a/colossalai/pipeline/__init__.py +++ b/colossalai/pipeline/__init__.py @@ -1,4 +1,4 @@ from .pipelinable import PipelinableContext, PipelinableModel -from .layer_sepc import LayerSpec +from .layer_spec import LayerSpec __all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec'] \ No newline at end of file diff --git a/colossalai/pipeline/layer_sepc.py b/colossalai/pipeline/layer_spec.py similarity index 100% rename from colossalai/pipeline/layer_sepc.py rename to colossalai/pipeline/layer_spec.py diff --git a/colossalai/pipeline/pipelinable.py b/colossalai/pipeline/pipelinable.py index 4d37c9833..9731530a6 100644 --- a/colossalai/pipeline/pipelinable.py +++ b/colossalai/pipeline/pipelinable.py @@ -9,7 +9,7 @@ from colossalai.nn.layer.utils import CheckpointModule from colossalai.tensor import ColoParameter from colossalai.core import global_context as gpc from colossalai.context import ParallelMode -from .layer_sepc import LayerSpec +from .layer_spec import LayerSpec class PipelinableContext(InsertPostInitMethodToModuleSubClasses): diff --git a/docs/colossalai/colossalai.pipeline.layer_sepc.rst b/docs/colossalai/colossalai.pipeline.layer_sepc.rst index 0ff6a83c2..156660b5c 100644 --- a/docs/colossalai/colossalai.pipeline.layer_sepc.rst +++ b/docs/colossalai/colossalai.pipeline.layer_sepc.rst @@ -1,5 +1,5 @@ colossalai.pipeline.layer\_sepc =============================== -.. automodule:: colossalai.pipeline.layer_sepc +.. automodule:: colossalai.pipeline.layer_spec :members: diff --git a/docs/colossalai/colossalai.pipeline.rst b/docs/colossalai/colossalai.pipeline.rst index adaebea2d..6f7652d49 100644 --- a/docs/colossalai/colossalai.pipeline.rst +++ b/docs/colossalai/colossalai.pipeline.rst @@ -8,6 +8,6 @@ colossalai.pipeline .. toctree:: :maxdepth: 2 - colossalai.pipeline.layer_sepc + colossalai.pipeline.layer_spec colossalai.pipeline.pipelinable colossalai.pipeline.utils