Skip to content

Commit 13b1efb

Browse files
committed
Updating test
1 parent 560b02d commit 13b1efb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

sycl/test/check_device_code/matrix/matrix-nvptx-tf32-test.cpp

+4-4
Original file line numberDiff line numberDiff line change
@@ -80,10 +80,10 @@ int main() {
8080
// CHECK: tail call i32 @llvm.nvvm.f2tf32.rna(float {{.*}}
8181
// Round a, b to tf32
8282
for (auto i = 0; i < 4; ++i)
83-
sub_a.data[i] = float_to_tf32(sub_a.data[i]);
83+
sub_a.data[i] = round_to_tf32(sub_a.data[i]);
8484

8585
for (auto i = 0; i < 4; ++i)
86-
sub_b.data[i] = float_to_tf32(sub_b.data[i]);
86+
sub_b.data[i] = round_to_tf32(sub_b.data[i]);
8787

8888
//CHECK: tail call { float, float, float, float, float, float, float, float } @llvm.nvvm.wmma.m16n16k8.mma.row.row.tf32(i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 %{{.*}}, i32 {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}) #{{.*}}
8989
sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);
@@ -125,10 +125,10 @@ int main() {
125125
// CHECK: tail call i32 @llvm.nvvm.f2tf32.rna(float {{.*}}
126126
// Round a, b to tf32
127127
for (auto i = 0; i < 4; ++i)
128-
sub_a.data[i] = float_to_tf32(sub_a.data[i]);
128+
sub_a.data[i] = round_to_tf32(sub_a.data[i]);
129129

130130
for (auto i = 0; i < 4; ++i)
131-
sub_b.data[i] = float_to_tf32(sub_b.data[i]);
131+
sub_b.data[i] = round_to_tf32(sub_b.data[i]);
132132

133133
//CHECK: tail call { float, float, float, float, float, float, float, float } @llvm.nvvm.wmma.m16n16k8.mma.col.col.tf32(i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, i32 {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}, float {{.*}}) #{{.*}}
134134
sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);

0 commit comments

Comments
 (0)