composable_kernel
composable_kernel copied to clipboard
Complex Contraction CK Bilinear Example
This is example of complex tensor contraction based on 4 GEMM. All imaginary and real tensor are explicitly defined.
Hi @ozturkosu why this example is called as complex? I see that you use real data types, not complex.
Hi @bartekxk , When it was implemented, CK did not support struct data type therefore image and real part are stored as separate tensor.