From bd53e98df32f0eff71556bdfb74543e1dda3c1d7 Mon Sep 17 00:00:00 2001 From: Wenkai Du Date: Fri, 22 Nov 2019 10:29:32 -0800 Subject: [PATCH] Fix build error with hip-clang --- src/common.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/common.cu b/src/common.cu index 07ebcd90a3..3edba38831 100644 --- a/src/common.cu +++ b/src/common.cu @@ -196,7 +196,7 @@ __device__ half testValue(const size_t offset, const int rep, const int ra } template<> __device__ rccl_bfloat16 testValue(const size_t offset, const int rep, const int rank) { - return (float)testValue(offset, rep, rank); + return rccl_bfloat16(testValue(offset, rep, rank)); } // Operations