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

File test_coloproxy.py

tests/test_fx/test_coloproxy.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2import torch.nn as nn
3from torch.fx import GraphModule
4

Callers

nothing calls this directly

Calls 1

test_coloproxyFunction · 0.85

Tested by

no test coverage detected