1. 14 7月, 2022 2 次提交
  2. 02 7月, 2022 1 次提交
    • L
      unify cpu context, part2 (#44012) · 755438a7
      Leo Chen 提交于
      * fix init()
      
      * delete test_device_context
      
      * replace CPUDeviceContext with CPUContext
      
      * fix test_scalar
      
      * remove dot_op.cc
      
      * fix compile
      755438a7
  3. 26 6月, 2022 1 次提交
  4. 30 11月, 2021 1 次提交
  5. 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