AI/rocm-systems
2
0
Fork 0
Koodi Ongelmat Pull-pyynnöt Actions Paketit Projektit Julkaisut Wiki Toiminta

Ensure backward compatibility for fp8 datatypes (#126)

Selaa lähdekoodia 
* Ensure backward compatibility for fp8 datatypes

Signed-off-by: ravaidya <ravaidya@amd.com>

* Update code comments

Signed-off-by: ravaidya <ravaidya@amd.com>

---------

Signed-off-by: ravaidya <ravaidya@amd.com>
This commit is contained in:
Rahul Vaidya
2025-05-15 13:56:40 -05:00
committed by GitHub
vanhempi 4b2b635766
commit 0abe3c80bb
2 muutettua tiedostoa jossa 8 lisäystä ja 8 poistoa
Download Patch File Download Diff File Expand all files Collapse all files
src/common.h
+4 -4
Näytä tiedosto
@@ -22,10 +22,10 @@
#include <fstream>
#include <iostream>
// Ensures backward compatibility for FP8 types in RCCL 2.24.3 and later
#if NCCL_VERSION_CODE >= NCCL_VERSION(2,24,3)
#define ncclFp8E4M3 ncclFloat8e4m3
#define ncclFp8E5M2 ncclFloat8e5m2
// Ensures backward compatibility for FP8 datatypes
#if NCCL_VERSION_CODE < NCCL_VERSION(2,24,3)
#define ncclFloat8e4m3 ncclFp8E4M3
#define ncclFloat8e5m2 ncclFp8E5M2
#endif
// For nccl.h < 2.13 since we define a weak fallback
verifiable/verifiable.cu
+4 -4
Näytä tiedosto
@@ -27,10 +27,10 @@
#else
#define HAVE_ncclfp8_HOST 1
#endif
// Ensures backward compatibility for FP8 types in RCCL 2.24.3 and later
#if NCCL_VERSION_CODE >= NCCL_VERSION(2,24,3)
#define ncclFp8E4M3 ncclFloat8e4m3
#define ncclFp8E5M2 ncclFloat8e5m2
// Ensures backward compatibility for FP8 datatypes
#if NCCL_VERSION_CODE < NCCL_VERSION(2,24,3)
#define ncclFloat8e4m3 ncclFp8E4M3
#define ncclFloat8e5m2 ncclFp8E5M2
#endif
#else
#define HAVE_ncclfp8 0