From a81b7cd5ea74423f6f507d1474d76143b02ac469 Mon Sep 17 00:00:00 2001
From: Dounia <dounia.khaldi@intel.com>
Date: Thu, 22 Feb 2024 08:44:19 -0800
Subject: [PATCH] [SYCL][Matrix tests] Missing general double type case in
 initialization and ref compute

---
 sycl/test-e2e/Matrix/common.hpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/sycl/test-e2e/Matrix/common.hpp b/sycl/test-e2e/Matrix/common.hpp
index 155cf012d9754..1e27fe6d7d989 100644
--- a/sycl/test-e2e/Matrix/common.hpp
+++ b/sycl/test-e2e/Matrix/common.hpp
@@ -64,7 +64,9 @@ void matrix_multiply_ref(Ta *A, Tb *B, Tc *C, int M, int N, int K,
             acc += make_fp32(va[i]) * make_fp32(vb[i]);
           else if constexpr (std::is_same_v<Ta, float> &&
                                  std::is_same_v<Tc, float> ||
-                             std::is_integral_v<Ta> && std::is_integral_v<Tc>)
+                             std::is_integral_v<Ta> && std::is_integral_v<Tc> ||
+                             (std::is_same_v<Ta, double> &&
+                              std::is_same_v<Tc, double>))
             acc += va[i] * vb[i];
           else if constexpr (std::is_same_v<Ta, sycl::half> &&
                              std::is_same_v<Tc, float>)
@@ -127,7 +129,8 @@ void matrix_rand(unsigned int rows, unsigned int cols, T *src, T val) {
 
   for (unsigned int i = 0; i < rows; i++) {
     for (unsigned int j = 0; j < cols; j++) {
-      if constexpr (std::is_same_v<T, bfloat16> || std::is_same_v<T, float>) {
+      if constexpr (std::is_same_v<T, bfloat16> || std::is_same_v<T, float> ||
+                    std::is_same_v<T, double>) {
         src[i * cols + j] = T(fdistr(dev));
       } else if constexpr (std::is_same_v<T, int8_t> ||
                            std::is_same_v<T, int32_t>) {