MCPcopy 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

1import torch
2
3from colossalai.fx import ColoTracer
4from colossalai.fx.passes.utils import assign_bfs_level_to_nodes, get_leaf, get_top

Callers

nothing calls this directly

Calls 1

test_graph_manipulationFunction · 0.85

Tested by

no test coverage detected