1. 02 7月, 2021 1 次提交
  2. 15 6月, 2021 1 次提交
    • J
      Support reduce_sum_op float16 (#32966) · 606939de
      jiangcheng 提交于
      * add reduce_sum_op by add self-kernel
      
      * set all ReduceKernel MPType for accuracy
      
      * add float16 test script which input is integer number
      
      * solve reduce sum float16 check_grad problem
      
      * solve conflict and change test script for CI
      
      * change kernel register for CI
      
      * remove all useless template
      606939de
  3. 27 5月, 2021 1 次提交
  4. 21 4月, 2021 1 次提交
    • W
      remove thrust include files (#32395) · ab6f8745
      wuhuanzhou 提交于
      * remove thrust includes, test=develop
      
      * fix compilation error, test=develop
      
      * fix compilation of truncated_gaussian_random_op, test=develop
      ab6f8745
  5. 08 3月, 2021 1 次提交
  6. 01 3月, 2021 1 次提交
  7. 04 12月, 2020 1 次提交
    • C
      Make transpose, trace, kron, reshape, sum op support complex type (#29321) · 879e913b
      chentianyu03 提交于
      * add complex64 and complex128 type; add +-*/@ and slice opreator for complex types
      
      * add test cases for complex elementwise, matmul and getitem unittest
      
      * add test cases for complex types
      
      * add test cases for complex matmul unittest
      
      * kron, reshape, transpose support complex types
      
      * sum and trace op support complex types
      
      * add test case of sum and trace op
      
      * fix the bug of imag part of complex not initialized
      
      * format file
      
      * format code style
      
      * kron support type promotion; modify test cases
      879e913b
  8. 13 7月, 2020 1 次提交
  9. 27 4月, 2020 1 次提交
    • L
      Add trace op (#23873) · 077e5a0f
      Li Fuchen 提交于
      * add trace op, test=develop
      
      * Optimized the implementation of trace op, test=develop
      
      * fix a bug of include in trace_op.h, test=develop
      
      * move trace API from creation to math, test=develop
      
      * modified en doc. test=develop
      
      * add complex trace api
      
      * add complex sum api, test=develop
      
      * modified en doc of complex sum and trace, test=develop
      
      * modified doc and trace API, test=develop
      
      * modified en doc of trace and sum, test=develop
      
      * modified comment in complex kron API, test=develop
      
      * OP Should Not Have Unused Input, test=develop
      
      * add GetExpectedKernelType, test=develop
      077e5a0f