SimpleJCublas.java

    public static IComplexNDArray gemm(IComplexNDArray A, IComplexNDArray B, IComplexNDArray C,
                                       float Alpha, float Beta) {

        JCublasNDArray cA = (JCublasNDArray) A;
        JCublasNDArray cB = (JCublasNDArray) B;
        JCublasNDArray cC = (JCublasNDArray) C;


        cuDoubleComplex alpha = cuDoubleComplex.cuCmplx(Alpha,0);
        cuDoubleComplex beta = cuDoubleComplex.cuCmplx(Beta,0);

        JCublas.cublasZgemm(
                'n', //trans
                'n',
                A.rows(),  // m
                B.columns(), // n
                B.rows(), //k,
                alpha,
                cA.pointer(), // A
                A.rows(),  // lda
                cB.pointer(), // x
                B.rows(), // incx
                beta,  // beta
                cC.pointer(), // y
                C.rows()); // incy


        cC.getData();


        return C;

    }