1. 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
  2. 24 3月, 2021 1 次提交
  3. 11 1月, 2021 1 次提交
  4. 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
  5. 15 12月, 2020 1 次提交
  6. 08 12月, 2020 1 次提交
  7. 03 12月, 2020 2 次提交
  8. 17 11月, 2020 1 次提交
  9. 25 9月, 2020 1 次提交
  10. 22 8月, 2020 1 次提交