1. 20 1月, 2022 1 次提交
  2. 30 11月, 2021 1 次提交
  3. 14 10月, 2021 1 次提交
    • Z
      Merge momentum ops/kernels (#36380) · f4eda869
      Zeng Jinle 提交于
      * merge momentum ops
      
      * update
      
      * add ut to improve coverage
      
      * remove optimizer change
      
      * fix error msg
      
      * update ut
      
      * add __restrict__ for CUDA
      
      * update ut
      
      * move merged_momentum_op to optimizer dir
      
      * fix coverage
      f4eda869