• 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
reduce_sum_op.cu 2.8 KB