ColossalAI/tests/test_fx/test_graph_manipulation.py at e99edfcb51df48dec17498c60bbbd06baa293c22

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-03 13:16:53 +00:00

Files

YuliangLiu0306 97d713855a [fx] methods to get fx graph property. (#1246 )

* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* manipulation

* [fx]add graph manipulation methods.

* [fx]methods to get fx graph property.

* add unit test

* add docstring to explain top node and leaf node in this context

2022-07-12 14:10:37 +08:00

1.5 KiB

Raw Blame History

View Raw

1.5 KiB Raw Blame History

1.5 KiB

Raw Blame History