github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-15 06:00:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[autoparallel] implemented all matmul strategy generator (#1650)

Browse Source
This commit is contained in:
Frank Lee
2022-09-27 12:06:25 +08:00
committed by GitHub
parent 03978aad45
commit 30e50c8b4a
8 changed files with 440 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_fx/test_tracer/test_torchrec_model/test_deepfm_model.py
View File

@@ -1,6 +1,3 @@
from curses import meta
from math import dist
from xml.dom import HierarchyRequestErr
from colossalai.fx.tracer import meta_patch
from colossalai.fx.tracer.tracer import ColoTracer
from colossalai.fx.tracer.meta_patch.patched_function import python_ops

3
tests/test_fx/test_tracer/test_torchrec_model/test_dlrm_model.py
View File

@@ -1,6 +1,3 @@
from curses import meta
from math import dist
from xml.dom import HierarchyRequestErr
from colossalai.fx.tracer import meta_patch
from colossalai.fx.tracer.tracer import ColoTracer
from colossalai.fx.tracer.meta_patch.patched_function import python_ops