Skip to content

Commit 077e0f4

Browse files
author
Hugh Delaney
committed
Changing names
1 parent b03e661 commit 077e0f4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

sycl/include/sycl/ext/oneapi/matrix/matrix-tensorcore.hpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -604,7 +604,9 @@ joint_matrix_mad(
604604
#endif // defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
605605
}
606606

607-
float float_to_tf32(float a) {
607+
// This function rounds the bottom 13 bits up or down, and then zeros out the
608+
// bottom bits
609+
float round_to_tf32(float a) {
608610
#if defined(__SYCL_DEVICE_ONLY__) && defined(__NVPTX__)
609611
int32_t tmp_int = __nvvm_f2tf32_rna(a);
610612
return __nvvm_bitcast_i2f(tmp_int);
@@ -618,7 +620,7 @@ float float_to_tf32(float a) {
618620
}
619621

620622
// This function just zeros out the bottom 13 bits of the tf32 type
621-
float tf32_to_float(float a) {
623+
float truncate_to_tf32(float a) {
622624
uint32_t tmp_uint = reinterpret_cast<uint32_t &>(a);
623625
tmp_uint &= 0xFFFFE000u;
624626
float ret = reinterpret_cast<float &>(tmp_uint);

0 commit comments

Comments
 (0)