- 20 May, 2022 1 commit
-
-
ver217 authored
* refactor parallel action * polish unit tests
-
- 19 May, 2022 2 commits
-
-
ver217 authored
* polish test_gpt * update op unit tests * update test model
-
ver217 authored
* 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
-
- 16 May, 2022 1 commit
-
-
ver217 authored
* derive compute pattern from dist spec * polish code
-
- 13 May, 2022 1 commit
-
-
ver217 authored
* add dist spec * update linear op * polish code * polish code * update embedding op * polish unit tests * polish unit tests * polish comments * polish code * add test_dist_spec_mgr * polish code * refactor folder structure * polish unit tests * add get_process_group() for TensorSpec * polish code
-
- 06 May, 2022 1 commit
-
-
Ziyue Jiang authored
-
- 28 Apr, 2022 2 commits
-
-
Ziyue Jiang authored
-
Ziyue Jiang authored
* refine linear and add function to ColoTensor * add gather for layernorm * polish * polish
-
- 27 Apr, 2022 1 commit
-
-
Ziyue Jiang authored
-
- 26 Apr, 2022 1 commit
-
-
Ziyue Jiang authored
-
- 24 Apr, 2022 4 commits
-
-
Ziyue Jiang authored
-
Jiarui Fang authored
-
Ziyue Jiang authored
-
Ziyue Jiang authored
-
- 22 Apr, 2022 2 commits
-
-
Jiarui Fang authored
This reverts commit ac88de6d.
-
Jiarui Fang authored
* revert zero tensors back * [tensor] init row 1d linear
-