1. 19 May, 2022 1 commit
    • ver217's avatar
      [tensor] refactor colo-tensor (#992) · ad536e30
      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
      ad536e30
  2. 16 May, 2022 1 commit
  3. 13 May, 2022 1 commit
    • 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
  4. 09 May, 2022 1 commit