github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-25 03:31:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx] Add linear metainfo class for auto parallel (#1783)

Browse Source 
* [fx] metainfo class for auto parallel

* [fx] add unit test for linear metainfo

* [fx] fix bwd param for linear

* [fx] modify unit test

* [fx] modify unit test

* [fx] modify import

* [fx] modify import

* [fx] modify import

* [fx] move meta profiler to auto parallel
This commit is contained in:
Boyuan Yao
2022-11-04 10:55:09 +08:00
committed by GitHub
parent e8a9bebc87
commit 05ce3d369f
10 changed files with 516 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/auto_parallel/meta_profiler/__init__.py Normal file
View File

@@ -0,0 +1,3 @@
from .meta_registry import *
from .metainfo import *
from .registry import meta_register