SIGN IN SIGN UP
hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 96 Python

[fx]Split partition with DAG information (#2025)

* add DAG to split_module

* add comment

* add test case for DAG

* remove print

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
Z
Ziyue Jiang committed
632753abbc62f15638f8595d0822e138c02fd684
Parent: ea0f6b8
Committed by GitHub <noreply@github.com> on 11/25/2022, 9:42:48 AM