1. 04 5月, 2022 1 次提交
  2. 01 5月, 2022 1 次提交
  3. 25 4月, 2022 1 次提交
  4. 23 4月, 2022 1 次提交
  5. 22 4月, 2022 1 次提交
    • M
      [WIP] Algorithm Cache of cuBlasLt Epilogue (#41010) · 19650d72
      Ming-Xu Huang 提交于
      * Fix leading dimension setting error in fused_gemm_epilogue_grad_op.
      
      * Add dyload to cuBlasLt functions.
      
      * Added cublasLtMatmulAlgoGetHeuristic to improve performance.
      
      * Added FLAGS_cublaslt_exhaustive_search_times to cublasLt epilogue
      
      * Added UTs to FLAGS_cublaslt_exhaustive_search_times
      
      * Added warmup runs in algo searching of Gemm epilogue.
      
      * Update copyright and documents.
      
      * Fixed error handling.
      19650d72
  6. 20 4月, 2022 1 次提交
  7. 19 4月, 2022 3 次提交
  8. 18 4月, 2022 3 次提交
  9. 15 4月, 2022 3 次提交
  10. 14 4月, 2022 4 次提交
  11. 13 4月, 2022 5 次提交
  12. 12 4月, 2022 3 次提交
  13. 11 4月, 2022 2 次提交
  14. 09 4月, 2022 1 次提交
  15. 08 4月, 2022 2 次提交
  16. 07 4月, 2022 5 次提交
  17. 06 4月, 2022 1 次提交
  18. 03 4月, 2022 1 次提交
    • F
      add maximum limit for grid of index_select (#41127) · af8d2482
      FlyingQianMM 提交于
      * limit grid dim for index select
      
      * mv LimitGridDim into gpu_launch_config.h
      
      * fix conflicts
      
      * fix conflicts
      
      * fix code style
      
      * set block to 256
      
      * fix grid setting
      
      * set dtype of block_dim to unsigned int
      af8d2482
  19. 01 4月, 2022 1 次提交
    • W
      [Eager] Support pinned (#41035) · f3270fc8
      wanghuancoder 提交于
      * support pinned, test=develop
      
      * support async_write, test=develop
      
      * refine, test=develop
      
      * refine, test=develop
      
      * refine, test=develop
      
      * refine,test=develop
      
      * refine, test=develop
      
      * refine, test=develop
      
      * refine, test=develop
      
      * refine, test=develop
      f3270fc8