ColossalAI/tests/test_tensor/test_context.py at 4d8a574cd3beb18e497d854e599f2899bc8afa70

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-11 18:14:16 +00:00

Files

ver217 ad536e308e [tensor] refactor colo-tensor (#992 )

* refactor colo-tensor and update linear op

* polish code

* polish code

* update ops and unit tests

* update unit tests

* polish code

* rename dist_spec module

* polish code

* polish code

* remove unneeded import

* fix pipelinable

2022-05-19 12:44:59 +08:00

1.0 KiB

Raw Blame History

View Raw

1.0 KiB Raw Blame History

1.0 KiB

Raw Blame History