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

File test_fp8_cast.py

tests/test_fp8/test_fp8_cast.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2from torch.testing import assert_close
3
4from colossalai.accelerator import get_accelerator

Callers

nothing calls this directly

Calls 1

test_fp8_castFunction · 0.85

Tested by

no test coverage detected