MCPcopy
hub / github.com/hpcaitech/ColossalAI / test_solver.py

File test_solver.py

tests/test_auto_parallel/test_offload/test_solver.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import pytest
2import torch.fx
3from torch.fx import GraphModule
4from torch.utils._pytree import tree_map

Callers

nothing calls this directly

Calls 1

solver_testFunction · 0.85

Tested by

no test coverage detected