- 22 Jun, 2022 1 commit
-
-
ver217 authored
* dist spec s2s uses all-to-all * update unit test * add sanity check * polish unitest test with titans * add sanity check for DistMgr * add sanity check Co-authored-by:jiaruifang <fangjiarui123@gmail.com>
-
- 31 May, 2022 1 commit
-
-
ver217 authored
-
- 19 May, 2022 1 commit
-
-
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
-
- 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
-