ColossalAI/tests/test_tensor/test_linear_tp.py at 4df6471f5d044150edfedfe1cf8370a604e096ca - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-09 04:46:44 +00:00

Files

Ziyue Jiang 26d4ab8b03 [Tensor] Add function to spec and update linear 1Drow and unit tests (#869 )

2022-04-26 10:15:26 +08:00

3.2 KiB

Raw Blame History

View Raw