Skip to content
This repository was archived by the owner on Mar 28, 2023. It is now read-only.

Commit 473f88b

Browse files
committed
format
Signed-off-by: JackAKirk <[email protected]>
1 parent 32eac9a commit 473f88b

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

SYCL/Matrix/element_wise_wi_marray.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
//===----------------------------------------------------------------------===//
88
// REQUIRES: cuda
99

10-
// RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple -Xsycl-target-backend --cuda-gpu-arch=sm_80 -DSYCL_EXT_ONEAPI_MATRIX=3 %s -o %t.out
10+
// RUN: %clangxx -fsycl -fsycl-targets=%sycl_triple -Xsycl-target-backend --cuda-gpu-arch=sm_80 -DSYCL_EXT_ONEAPI_MATRIX=3 %s -o %t.out
1111
// RUN: %t.out
1212

1313
#include <sycl/sycl.hpp>

SYCL/Matrix/joint_matrix_tensorcore.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -195,15 +195,15 @@ void test(queue &q) {
195195
accB.get_pointer() + (k * K * Big_N) + (n * N),
196196
Big_N);
197197

198-
// round values to correct precision if using tf32
199-
if constexpr (std::is_same<T3, precision::tf32>::value) {
200-
auto wi_size = sub_a.wi_marray.size();
201-
assert(wi_size == sub_b.wi_marray.size());
202-
for (auto i = 0; i < wi_size; ++i) {
203-
sub_a.wi_marray[i] = round_to_tf32(sub_a.wi_marray[i]);
204-
sub_b.wi_marray[i] = round_to_tf32(sub_b.wi_marray[i]);
198+
// round values to correct precision if using tf32
199+
if constexpr (std::is_same<T3, precision::tf32>::value) {
200+
auto wi_size = sub_a.wi_marray.size();
201+
assert(wi_size == sub_b.wi_marray.size());
202+
for (auto i = 0; i < wi_size; ++i) {
203+
sub_a.wi_marray[i] = round_to_tf32(sub_a.wi_marray[i]);
204+
sub_b.wi_marray[i] = round_to_tf32(sub_b.wi_marray[i]);
205+
}
205206
}
206-
}
207207

208208
sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);
209209
}

0 commit comments

Comments
 (0)