Fix build error with hip-clang

2019-11-22 10:29:32 -08:00
parent 1cda2f52b6
commit bd53e98df3
1 changed files with 1 additions and 1 deletions
@@ -196,7 +196,7 @@ __device__ half testValue<half>(const size_t offset, const int rep, const int ra
 }
 template<>
 __device__ rccl_bfloat16 testValue<rccl_bfloat16>(const size_t offset, const int rep, const int rank) {
-  return (float)testValue<float>(offset, rep, rank);
+  return rccl_bfloat16(testValue<float>(offset, rep, rank));
 }

 // Operations