[SYCL][Matrix] add tests for tf32 type

SYCL/Matrix/XMX8/element_wise_all_ops_tf32.cpp

@@ -0,0 +1,27 @@
+//==----------- element_wise_all_ops_tf32.cpp  - DPC++ joint_matrix---------==//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+// REQUIRES: matrix-xmx8
+
+// RUN: %clangxx -fsycl %s -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// XFAIL:*
+
+#include <iostream>
+#include <random>
+#include <sycl/sycl.hpp>
+
+using namespace sycl;
+using namespace sycl::ext::intel;
+using namespace sycl::ext::oneapi;
+using namespace sycl::ext::oneapi::experimental::matrix;
+
+#define SG_SZ 8
+
+#include "../element_wise_all_ops_tf32_impl.hpp"

SYCL/Matrix/XMX8/joint_matrix_tf32.cpp

@@ -0,0 +1,27 @@
+//==---------------- joint_matrix_tf32.cpp  - DPC++ joint_matrix------------==//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+// REQUIRES: matrix-xmx8
+
+// RUN: %clangxx -fsycl %s -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// XFAIL:*
+
+#include <iostream>
+#include <random>
+#include <sycl/sycl.hpp>
+
+using namespace sycl;
+using namespace sycl::ext::intel;
+using namespace sycl::ext::oneapi;
+using namespace sycl::ext::oneapi::experimental::matrix;
+
+#define SG_SZ 8
+
+#include "../joint_matrix_tf32_impl.hpp"

SYCL/Matrix/element_wise_all_ops_tf32.cpp

@@ -0,0 +1,27 @@
+//==----------- element_wise_all_ops_tf32.cpp  - DPC++ joint_matrix---------==//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+// REQUIRES: matrix
+
+// RUN: %clangxx -fsycl %s -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// XFAIL:*
+
+#include <iostream>
+#include <random>
+#include <sycl/sycl.hpp>
+
+using namespace sycl;
+using namespace sycl::ext::intel;
+using namespace sycl::ext::oneapi;
+using namespace sycl::ext::oneapi::experimental::matrix;
+
+#define SG_SZ 16
+
+#include "element_wise_all_ops_tf32_impl.hpp"

SYCL/Matrix/element_wise_all_ops_tf32_impl.hpp

@@ -0,0 +1,239 @@
+
+#define TM 8
+#define TN SG_SZ
+#define TK 16
+
+template <typename T, size_t NUM_ROWS, size_t NUM_COLS> struct big_matrix {
+public:
+  T *mat;
+
+public:
+  T *get_data() { return mat; }
+  void set_data(T *data) { mat = data; }
+  big_matrix(T *data) : mat(data) {}
+};
+
+template <typename T, size_t M, size_t N>
+void assert_ops_ref(host_accessor<T, 2, access::mode::read> C,
+                    const float ref) {
+  for (size_t i = 0; i < M; i++)
+    for (size_t j = 0; j < N; j++) {
+      auto diff = C[i][j] - ref;
+      assert(std::fabs(static_cast<float>(diff)) <
+             std::numeric_limits<float>::epsilon());
+    }
+}
+template <typename T, typename Ts, size_t M, size_t N>
+void matrix_verify_add(queue q, big_matrix<Ts, M, N> &A, nd_range<2> &r,
+                       const float ref) {
+  buffer<Ts, 2> bufA(A.get_data(), range<2>(M, N));
+
+  q.submit([&](handler &cgh) {
+     sycl::accessor accA{bufA, cgh, sycl::read_write};
+
+     cgh.parallel_for<class add_matrix>(
+         r, [accA](nd_item<2> spmd_item) [[intel::reqd_sub_group_size(SG_SZ)]] {
+           const auto global_idx = spmd_item.get_global_id(0);
+           const auto global_idy = spmd_item.get_global_id(1);
+           const auto sg_startx = global_idx - spmd_item.get_local_id(0);
+           const auto sg_starty = global_idy - spmd_item.get_local_id(1);
+
+           sub_group sg = spmd_item.get_sub_group();
+           joint_matrix<sub_group, T, use::a, TM, TK, layout::row_major> sub_a;
+
+           joint_matrix_fill(sg, sub_a, round_to_tf32(5.0));
+
+           auto wi_slice_a =
+               ext::intel::experimental::matrix::get_wi_data(sg, sub_a);
+           for (int i = 0; i < wi_slice_a.length(); i++) {
+             wi_slice_a[i] = wi_slice_a[i] + 2;
+           }
+
+           ext::intel::experimental::matrix::joint_matrix_store(
+               sg, sub_a,
+               accA.get_pointer() + (sg_startx * TM) * N +
+                   sg_starty / SG_SZ * TN,
+               N);
+         }); // parallel for
+   }).wait();
+  assert_ops_ref<Ts, M, N>(bufA.get_host_access(sycl::read_only), ref);
+}
+
+template <typename T, typename Ts, size_t M, size_t N>
+void matrix_verify_sub(queue q, big_matrix<Ts, M, N> &A, nd_range<2> &r,
+                       const float ref) {
+  buffer<Ts, 2> bufA(A.get_data(), range<2>(M, N));
+
+  q.submit([&](handler &cgh) {
+     sycl::accessor accA{bufA, cgh, sycl::read_write};
+
+     cgh.parallel_for<class sub_matrix>(
+         r, [accA](nd_item<2> spmd_item) [[intel::reqd_sub_group_size(SG_SZ)]] {
+           const auto global_idx = spmd_item.get_global_id(0);
+           const auto global_idy = spmd_item.get_global_id(1);
+           const auto sg_startx = global_idx - spmd_item.get_local_id(0);
+           const auto sg_starty = global_idy - spmd_item.get_local_id(1);
+
+           sub_group sg = spmd_item.get_sub_group();
+           joint_matrix<sub_group, T, use::a, TM, TK, layout::row_major> sub_a;
+
+           joint_matrix_fill(sg, sub_a, round_to_tf32(5.0));
+
+           auto wi_slice_a =
+               ext::intel::experimental::matrix::get_wi_data(sg, sub_a);
+           for (int i = 0; i < wi_slice_a.length(); i++) {
+             wi_slice_a[i] = wi_slice_a[i] - round_to_tf32(2);
+           }
+           ext::intel::experimental::matrix::joint_matrix_store(
+               sg, sub_a,
+               accA.get_pointer() + (sg_startx * TM) * N +
+                   sg_starty / SG_SZ * TN,
+               N);
+         }); // parallel for
+   }).wait();
+  assert_ops_ref<Ts, M, N>(bufA.get_host_access(sycl::read_only), ref);
+}
+
+template <typename T, typename Ts, size_t M, size_t N>
+void matrix_verify_mul(queue q, big_matrix<Ts, M, N> &A, nd_range<2> &r,
+                       const float ref) {
+  buffer<Ts, 2> bufA(A.get_data(), range<2>(M, N));
+
+  q.submit([&](handler &cgh) {
+     sycl::accessor accA{bufA, cgh, sycl::read_write};
+
+     cgh.parallel_for<class mul_matrix>(
+         r, [accA](nd_item<2> spmd_item) [[intel::reqd_sub_group_size(SG_SZ)]] {
+           const auto global_idx = spmd_item.get_global_id(0);
+           const auto global_idy = spmd_item.get_global_id(1);
+           const auto sg_startx = global_idx - spmd_item.get_local_id(0);
+           const auto sg_starty = global_idy - spmd_item.get_local_id(1);
+
+           sub_group sg = spmd_item.get_sub_group();
+           joint_matrix<sub_group, T, use::a, TM, TK, layout::row_major> sub_a;
+           joint_matrix_fill(sg, sub_a, round_to_tf32(5.0));
+
+           auto wi_slice_a =
+               ext::intel::experimental::matrix::get_wi_data(sg, sub_a);
+           for (int i = 0; i < wi_slice_a.length(); i++) {
+             wi_slice_a[i] = wi_slice_a[i] * round_to_tf32(3.0);
+           }
+           ext::intel::experimental::matrix::joint_matrix_store(
+               sg, sub_a,
+               accA.get_pointer() + (sg_startx * TM) * N +
+                   sg_starty / SG_SZ * TN,
+               N);
+         }); // parallel for
+   }).wait();
+  assert_ops_ref<Ts, M, N>(bufA.get_host_access(sycl::read_only), ref);
+}
+
+template <typename T, typename Ts, size_t M, size_t N>
+void matrix_verify_div(queue q, big_matrix<Ts, M, N> &A, nd_range<2> &r,
+                       const float ref) {
+  buffer<Ts, 2> bufA(A.get_data(), range<2>(M, N));
+
+  q.submit([&](handler &cgh) {
+     sycl::accessor accA{bufA, cgh, sycl::read_write};
+
+     cgh.parallel_for<class div_matrix>(
+         r, [accA](nd_item<2> spmd_item) [[intel::reqd_sub_group_size(SG_SZ)]] {
+           const auto global_idx = spmd_item.get_global_id(0);
+           const auto global_idy = spmd_item.get_global_id(1);
+           const auto sg_startx = global_idx - spmd_item.get_local_id(0);
+           const auto sg_starty = global_idy - spmd_item.get_local_id(1);
+
+           sub_group sg = spmd_item.get_sub_group();
+           joint_matrix<sub_group, T, use::a, TM, TK, layout::row_major> sub_a;
+
+           joint_matrix_fill(sg, sub_a, round_to_tf32(4.0));
+
+           auto wi_slice_a =
+               ext::intel::experimental::matrix::get_wi_data(sg, sub_a);
+           for (int i = 0; i < wi_slice_a.length(); i++) {
+             wi_slice_a[i] = wi_slice_a[i] / round_to_tf32(2.0);
+           }
+           ext::intel::experimental::matrix::joint_matrix_store(
+               sg, sub_a,
+               accA.get_pointer() + (sg_startx * TM) * N +
+                   sg_starty / SG_SZ * TN,
+               N);
+         }); // parallel for
+   }).wait();
+  assert_ops_ref<Ts, M, N>(bufA.get_host_access(sycl::read_only), ref);
+}
+
+template <typename T, typename Ts, size_t M, size_t N>
+void matrix_verify_logic(queue q, big_matrix<Ts, M, N> &A, nd_range<2> &r,
+                         const float ref) {
+  buffer<Ts, 2> bufA(A.get_data(), range<2>(M, N));
+
+  q.submit([&](handler &cgh) {
+     sycl::accessor accA{bufA, cgh, sycl::read_write};
+     cgh.parallel_for<class logic_matrix>(
+         r, [accA](nd_item<2> spmd_item) [[intel::reqd_sub_group_size(SG_SZ)]] {
+           const auto global_idx = spmd_item.get_global_id(0);
+           const auto global_idy = spmd_item.get_global_id(1);
+           const auto sg_startx = global_idx - spmd_item.get_local_id(0);
+           const auto sg_starty = global_idy - spmd_item.get_local_id(1);
+
+           sub_group sg = spmd_item.get_sub_group();
+           joint_matrix<sub_group, T, use::a, TM, TK, layout::row_major> sub_a;
+
+           joint_matrix_fill(sg, sub_a, round_to_tf32(5.0));
+
+           auto wi_slice_a =
+               ext::intel::experimental::matrix::get_wi_data(sg, sub_a);
+           for (int i = 0; i < wi_slice_a.length(); i++) {
+             if (wi_slice_a[i]) {
+               if (wi_slice_a[i] > 2.0 || wi_slice_a[i] >= 2.0 ||
+                   wi_slice_a[i] < 2.0 || wi_slice_a[i] <= 2.0) {
+                 Ts val = (wi_slice_a[i] != 2.0) ? wi_slice_a[i] : 2.0;
+                 val = val - static_cast<float>(1);
+                 val = val + static_cast<float>(1);
+                 if (wi_slice_a[i] == 2.0) {
+                   val = val - static_cast<float>(2);
+                   val = val * static_cast<float>(3);
+                   val = val / static_cast<float>(2);
+
+                 } else {
+                   val = val + static_cast<float>(2);
+                 }
+                 wi_slice_a[i] = val;
+               }
+             }
+           }
+           ext::intel::experimental::matrix::joint_matrix_store(
+               sg, sub_a,
+               accA.get_pointer() + (sg_startx * TM) * N +
+                   sg_starty / SG_SZ * TN,
+               N);
+         }); // parallel for
+   }).wait();
+  assert_ops_ref<Ts, M, N>(bufA.get_host_access(sycl::read_only), ref);
+}
+
+static constexpr size_t MATRIX_M = TM * 2;
+static constexpr size_t MATRIX_N = TN * 2;
+float A[MATRIX_M][MATRIX_N];
+float D[MATRIX_M][MATRIX_N];
+
+int main() {
+
+  big_matrix<float, MATRIX_M, MATRIX_N> MD((float *)&D);
+  big_matrix<float, MATRIX_M, MATRIX_N> MA((float *)&A);
+
+  size_t NDRangeM = MATRIX_M / TM;
+  size_t NDRangeN = MATRIX_N / TN;
+  queue q;
+  nd_range<2> r({NDRangeM, NDRangeN * SG_SZ}, {1, 1 * SG_SZ});
+
+  matrix_verify_add<precision::tf32, float, MATRIX_M, MATRIX_N>(q, MA, r, 7.0);
+  matrix_verify_sub<precision::tf32, float, MATRIX_M, MATRIX_N>(q, MA, r, 3.0);
+  matrix_verify_mul<precision::tf32, float, MATRIX_M, MATRIX_N>(q, MA, r, 15.0);
+  matrix_verify_div<precision::tf32, float, MATRIX_M, MATRIX_N>(q, MA, r, 2.0);
+  matrix_verify_logic<precision::tf32, float, MATRIX_M, MATRIX_N>(q, MA, r,
+                                                                  7.0);
+
+  return 0;
+}

SYCL/Matrix/joint_matrix_tf32.cpp

@@ -0,0 +1,27 @@
+//==---------------- joint_matrix_tf32.cpp  - DPC++ joint_matrix------------==//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+// REQUIRES: matrix
+
+// RUN: %clangxx -fsycl %s -o %t.out -DSYCL_EXT_ONEAPI_MATRIX_VERSION=4
+// RUN: %CPU_RUN_PLACEHOLDER %t.out
+// RUN: %GPU_RUN_PLACEHOLDER %t.out
+
+// XFAIL:*
+
+#include <iostream>
+#include <random>
+#include <sycl/sycl.hpp>
+
+using namespace sycl;
+using namespace sycl::ext::intel;
+using namespace sycl::ext::oneapi;
+using namespace sycl::ext::oneapi::experimental::matrix;
+
+#define SG_SZ 16
+
+#include "joint_matrix_tf32_impl.hpp"