fix bug

Author: chansoonlim

ctmd/linalg/matmul.hpp

@@ -68,7 +68,7 @@ inline constexpr void matmul_impl(const in1_t &in1, const in2_t &in2,
             const auto ein2 =
                 core::eigen::to_eigen(in2).template cast<common_t>();
             auto eout = core::eigen::to_eigen(out);
-            eout = ein1 * ein2;
+            eout = (ein1 * ein2).template cast<typename out_t::value_type>();
             return;
         }
     }

ctmd/linalg/matvec.hpp

@@ -65,7 +65,7 @@ inline constexpr void matvec_impl(const in1_t &in1, const in2_t &in2,
             const auto ein2 =
                 core::eigen::to_eigen(in2).template cast<common_t>();
             auto eout = core::eigen::to_eigen(out);
-            eout = ein1 * ein2;
+            eout = (ein1 * ein2).template cast<typename out_t::value_type>();
             return;
         }
     }

ctmd/linalg/vecmat.hpp

@@ -65,7 +65,7 @@ inline constexpr void vecmat_impl(const in1_t &in1, const in2_t &in2,
             const auto ein2 =
                 core::eigen::to_eigen(in2).template cast<common_t>();
             auto eout = core::eigen::to_eigen(out);
-            eout = ein1 * ein2;
+            eout = (ein1 * ein2).template cast<typename out_t::value_type>();
             return;
         }
     }

tests/linalg/matmul/main.cpp

@@ -38,3 +38,17 @@ TEST(heap, matmul) {
 
     ASSERT_TRUE(allclose);
 }
+
+TEST(test, mixed) {
+    using T1 = int;
+    using T2 = float;
+    using T3 = double;
+
+    const auto a = md::mdarray<T1, md::dims<2>>{std::vector<T1>{1, 2, 3, 4},
+                                                md::dims<2>{2, 2}};
+    const auto b = md::mdarray<T2, md::dims<2>>{std::vector<T2>{5, 6, 7, 8},
+                                                md::dims<2>{2, 2}};
+    auto c = md::mdarray<T3, md::dims<2>>{md::dims<2>{2, 2}};
+
+    md::linalg::matmul(a, b, c);
+}