1. 19 2月, 2022 1 次提交
  2. 16 2月, 2022 1 次提交
    • L
      [bf16] pten matmul cuda kernel support bf16 (#39485) · d5a0d31a
      Leo Chen 提交于
      * pten matmul cuda kernel support bf16
      
      * fix pten kernel name
      
      * add matmul_grad bf16 kernel
      
      * add emptylike bf16 kernel
      
      * fix compile
      
      * suppport rocm
      
      * fix error
      
      * fix rocm
      
      * add bf16 header file
      
      * fix compile
      d5a0d31a
  3. 24 3月, 2021 1 次提交
  4. 11 1月, 2021 1 次提交
  5. 31 12月, 2020 1 次提交
    • C
      complex gradient matmul (#29966) · e012930a
      chentianyu03 提交于
      * dot op support complex types
      
      * matmul support complex types
      
      * add test case
      
      * matmul broadcast gradient support complex
      
      * move conjFunctor to complex_functor.h
      e012930a
  6. 15 12月, 2020 1 次提交
  7. 08 12月, 2020 1 次提交
  8. 03 12月, 2020 2 次提交
  9. 17 11月, 2020 1 次提交
  10. 25 9月, 2020 1 次提交
  11. 22 8月, 2020 1 次提交