Add complex matmul test

Author: emmatyping

src/linalg/impl_linalg.rs

@@ -416,7 +416,7 @@ fn mat_mul_impl<A>(
             rhs_trans = CblasTrans;
         }
 
-        macro_rules! cast_ty {
+        macro_rules! gemm_scalar_cast {
             (f32, $var:ident) => {
                 cast_as(&$var)
             };
@@ -460,17 +460,17 @@ fn mat_mul_impl<A>(
                             CblasRowMajor,
                             lhs_trans,
                             rhs_trans,
-                            m as blas_index,               // m, rows of Op(a)
-                            n as blas_index,               // n, cols of Op(b)
-                            k as blas_index,               // k, cols of Op(a)
-                            cast_ty!($ty, alpha),          // alpha
-                            lhs_.ptr.as_ptr() as *const _, // a
-                            lhs_stride,                    // lda
-                            rhs_.ptr.as_ptr() as *const _, // b
-                            rhs_stride,                    // ldb
-                            cast_ty!($ty, beta),           // beta
-                            c_.ptr.as_ptr() as *mut _,     // c
-                            c_stride,                      // ldc
+                            m as blas_index,                 // m, rows of Op(a)
+                            n as blas_index,                 // n, cols of Op(b)
+                            k as blas_index,                 // k, cols of Op(a)
+                            gemm_scalar_cast!($ty, alpha),   // alpha
+                            lhs_.ptr.as_ptr() as *const _,   // a
+                            lhs_stride,                      // lda
+                            rhs_.ptr.as_ptr() as *const _,   // b
+                            rhs_stride,                      // ldb
+                            gemm_scalar_cast!($ty, beta),    // beta
+                            c_.ptr.as_ptr() as *mut _,       // c
+                            c_stride,                        // ldc
                         );
                     }
                     return;

xtest-blas/tests/oper.rs

@@ -313,6 +313,31 @@ fn gemm_c32_1_f() {
     );
 }
 
+#[test]
+fn gemm_c64_actually_complex() {
+    let mut a = range_mat_complex64(4,4);
+    a = a.map(|&i| if i.re > 8. { i.conj() } else { i });
+    let mut b = range_mat_complex64(4,6);
+    b = b.map(|&i| if i.re > 4. { i.conj() } else {i});
+    let mut y = range_mat_complex64(4,6);
+    let alpha = Complex64::new(0., 1.0);
+    let beta = Complex64::new(1.0, 1.0);
+    let answer = alpha * reference_mat_mul(&a, &b) + beta * &y;
+    general_mat_mul(
+        alpha.clone(),
+        &a,
+        &b,
+        beta.clone(),
+        &mut y,
+    );
+    assert_relative_eq!(
+        y.mapv(|i| i.norm_sqr()),
+        answer.mapv(|i| i.norm_sqr()),
+        epsilon = 1e-12,
+        max_relative = 1e-7
+    );
+}
+
 #[test]
 fn gen_mat_vec_mul() {
     let alpha = -2.3;