replacing rccl_bfloat16 with hip_bfloat16 (#1126)

Co-authored-by: mberenjk <mberenjk@amd.com>
2024-04-11 11:30:37 -05:00
parent d3899c0581
commit 428837ffe4
12 changed files with 30 additions and 306 deletions
@@ -8,7 +8,7 @@
 #include "ErrCode.hpp"
 #include "rccl/rccl.h"
 #include "rccl_float8.h"
-#include "rccl_bfloat16.h"
+#include <hip/hip_bfloat16.h>
 #include "hip/hip_fp16.h"

 namespace RcclUnitTesting
@@ -48,7 +48,7 @@ namespace RcclUnitTesting
    float*         F4; // ncclFloat32
    double*        F8; // ncclFloat64
    rccl_bfloat8*  B1; // ncclFp8E5M2
-    rccl_bfloat16* B2; // ncclBfloat16
+    hip_bfloat16*  B2; // ncclBfloat16

    constexpr PtrUnion() : ptr(nullptr) {}