AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

replacing rccl_bfloat16 with hip_bfloat16 (#1126)

Browse Source 
Co-authored-by: mberenjk <mberenjk@amd.com>
This commit is contained in:
mberenjkandmberenjk authored and GitHub committed 2024-04-11 11:30:37 -05:00
1 parent d3899c0581
commit 428837ffe4
12 files changed
+30 -306
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/include/device.h
+1 -1
View File
@@ -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"