1. 14 10月, 2021 2 次提交
  2. 13 10月, 2021 1 次提交
  3. 09 10月, 2021 1 次提交
  4. 08 10月, 2021 1 次提交
  5. 07 10月, 2021 1 次提交
  6. 06 10月, 2021 1 次提交
  7. 02 10月, 2021 1 次提交
  8. 01 10月, 2021 1 次提交
  9. 29 9月, 2021 1 次提交
  10. 25 9月, 2021 1 次提交
  11. 24 9月, 2021 1 次提交
  12. 08 9月, 2021 1 次提交
    • Masaki Kozuki's avatar
      enable ninja (#1164) · 9ce0a10f
      Masaki Kozuki 创作于
      - passing include directories to `CUDAExtension`'s `include_dirs` argument
      - removing `-I/path/to/dir` arguments from `extra_compile_args`
      9ce0a10f
  13. 04 9月, 2021 1 次提交
    • Burc Eryilmaz's avatar
      fix CUBLAS guards (#1162) · 54b93919
      Burc Eryilmaz 创作于
      
      
      * support for fused dense layer with cublasLt, fusion in both fprop and bprop
      
      * fix typo causing syntax error
      
      * add fused GEMM+gelu+GEMM modue
      
      * fix typo for workspace size
      
      * update cublas check for 11600
      
      * add tests for fused dense layer
      
      * fix CUDA 10.x path
      
      * safer guard around CUBLAS constants, remove unreferenced variable
      
      * more guard changes
      
      * guard against cublas version instead of cuda
      Co-authored-by: default avatarSukru Eryilmaz <seryilmaz@computelab-dgx1v-32.nvidia.com>
      54b93919
  14. 03 9月, 2021 2 次提交
  15. 02 9月, 2021 13 次提交
  16. 01 9月, 2021 3 次提交
  17. 31 8月, 2021 3 次提交
  18. 30 8月, 2021 1 次提交
  19. 21 8月, 2021 1 次提交
  20. 17 7月, 2021 3 次提交
    • Nan Zheng's avatar
      Added more fusion and vectorized kernel for transducer (#1125) · 0c2c6eea
      Nan Zheng 创作于
      * Added support for fused ReLU and dropout into transducer joint
      
      * Reorganized code selection path in transducer joint fwd
      * Added support for fused ReLU+dropout into transducer joint
      
      * Vectorize transducer loss backward with fused softmax (#3)
      
      * Nanz/transducer loss (#4)
      
      * Vectorize transducer loss backward with fused softmax
      
      * Added a predicate to avoid potential IMA
      
      * Nanz/transducer loss (#5)
      
      * Vectorize transducer loss backward with fused softmax
      
      * Added a predicate to avoid potentional IMA
      
      * Added more predicates to avoid IMAs
      
      * Updated documentations for newly added features.
      
      * Fixed a error in transducer.py
      0c2c6eea
    • yjk21's avatar
      Adds small-batch kernels (#1126) · ed719967
      yjk21 创作于
      ed719967
    • X Wang's avatar
      local_rank fix (#1129) · c1378e6f
      X Wang 创作于
      * local_rank and install cuda version fix
      c1378e6f