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