1. 16 8月, 2021 14 次提交
  2. 14 8月, 2021 1 次提交
  3. 13 8月, 2021 11 次提交
  4. 12 8月, 2021 11 次提交
  5. 11 8月, 2021 3 次提交
    • J
      [oneDNN] Fix to issue #34554 (#34623) · 0a5c99e8
      Jacek Czaja 提交于
      * - Added softmax without caching
      
      * - Binary is no longer manually cached
      
      * - Activation onednn caching removed
      
      * - Removed manual caching of activation
      
      * - modified UT
      
      * - fix
      
      * - fix
      
      * - fixes to building
      
      * - fix
      
      * - fix
      
      * - fix to UT
      
      * - Faulty UT workaround
      
      * - approval workaround
      
      * - Fixes after review
      
      * - compilation fixes
      
      * - more lint fixes
      
      * - more fixes after review
      
      * - fixes after another round of review
      0a5c99e8
    • Z
      [AMP] add state_dict and load_state_dict and unittest for class GradScaler (#34300) · 99f8f5c8
      zhangbo9674 提交于
      * add state_dict and load_state_dict and unittest for class GradScaler
      
      * refine unittest for coverage of load_state_dict
      
      * refine comments of code-block
      
      * refine some comments
      
      * refine state_dict code and unittest
      
      * add #require gpu, xpu for GradScaler get/set example code
      
      * add #require gpu, xpu for GradScaler get/set example code
      
      * refine example code
      
      * refine unittest for state_dict
      
      * refine unittest for state_dict
      
      * fix bug of DataLoader in TestGradScalerStateDict
      
      * add flag FLAGS_cudnn_deterministic
      99f8f5c8
    • W
      `set_value_grad` propagate gradients to `Input` and `TensorValue` (#34304) · 9d02313c
      WeiXin 提交于
      * add set_value_grad op
      
      * add unittest.
      
      * polish unittest.
      
      * polish code.
      
      * support cuda kernel
      
      * polish code according to CI
      
      * polish code.
      
      * polish code
      
      * remove *.pyc
      
      * polish code.
      
      * add unittest to improve coverage.
      
      * polish code.
      9d02313c