Code
Hub
Workspaces
Connect
Indexed graphs
Engine
MCP
copy
Index your code
hub
/
github.com/hpcaitech/ColossalAI
/ test_graph_manipulation.py
File
test_graph_manipulation.py
tests/test_fx/test_graph_manipulation.py:None–None ·
view source on GitHub ↗
Source
from the content-addressed store, hash-verified
1
import
torch
2
3
from
colossalai.fx
import
ColoTracer
4
from
colossalai.fx.passes.utils
import
assign_bfs_level_to_nodes, get_leaf, get_top
Callers
nothing calls this directly
Calls
1
test_graph_manipulation
Function · 0.85
Tested by
no test coverage detected