github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix file name (#1759) 

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
pull/1760/head
Ziyue Jiang 2022-10-25 16:48:48 +08:00 committed by GitHub
parent 314d8c497f
commit 63f250bbd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/pipeline/__init__.py
View File

@ -1,4 +1,4 @@
from .pipelinable import PipelinableContext, PipelinableModel from .pipelinable import PipelinableContext, PipelinableModel
from .layer_sepc import LayerSpec from .layer_spec import LayerSpec
__all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec'] __all__ = ['PipelinableModel', 'PipelinableContext', 'LayerSpec']

0
colossalai/pipeline/layer_sepc.py → colossalai/pipeline/layer_spec.py
View File

2
colossalai/pipeline/pipelinable.py
View File

@ -9,7 +9,7 @@ from colossalai.nn.layer.utils import CheckpointModule
from colossalai.tensor import ColoParameter from colossalai.tensor import ColoParameter
from colossalai.core import global_context as gpc from colossalai.core import global_context as gpc
from colossalai.context import ParallelMode from colossalai.context import ParallelMode
from .layer_sepc import LayerSpec from .layer_spec import LayerSpec
class PipelinableContext(InsertPostInitMethodToModuleSubClasses): class PipelinableContext(InsertPostInitMethodToModuleSubClasses):

2
docs/colossalai/colossalai.pipeline.layer_sepc.rst
View File

@ -1,5 +1,5 @@
colossalai.pipeline.layer\_sepc colossalai.pipeline.layer\_sepc
=============================== ===============================
.. automodule:: colossalai.pipeline.layer_sepc .. automodule:: colossalai.pipeline.layer_spec
:members: :members:

2
docs/colossalai/colossalai.pipeline.rst
View File

@ -8,6 +8,6 @@ colossalai.pipeline
.. toctree:: .. toctree::
:maxdepth: 2 :maxdepth: 2
colossalai.pipeline.layer_sepc colossalai.pipeline.layer_spec
colossalai.pipeline.pipelinable colossalai.pipeline.pipelinable
colossalai.pipeline.utils colossalai.pipeline.utils