Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • ver217's avatar
    [tensor] design DistSpec and DistSpecManager for ColoTensor (#934) · 67c33f57
    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
    67c33f57
test_linear_tp.py 3 KB