replacing rccl_bfloat16 with hip_bfloat16 (#1126)

Co-authored-by: mberenjk <mberenjk@amd.com> [ROCm/rccl commit: 428837ffe4]
2024-04-11 11:30:37 -05:00
parent 165d51b255
commit da835cff9c
12 changed files with 30 additions and 306 deletions
@@ -11,7 +11,7 @@

 #include "nccl.h"
 #include "rccl_float8.h"
-#include "rccl_bfloat16.h"
+#include <hip/hip_bfloat16.h>
 #include "nccl_common.h"
 #include "align.h"
 #include "collectives.h"