introduce support for ncclFloat16/half in UT

2022-08-23 19:08:38 +00:00
parent 3c1b1ec8c8
commit f6e00dec13
8 changed files with 19 additions and 9 deletions
@@ -2,6 +2,7 @@
 #include "ErrCode.hpp"
 #include "rccl/rccl.h"
 #include "rccl_bfloat16.h"
+#include "hip/hip_fp16.h"

 namespace RcclUnitTesting
 {
@@ -35,6 +36,7 @@ namespace RcclUnitTesting
    uint32_t*      U4; // ncclUint32
    int64_t*       I8; // ncclInt64
    uint64_t*      U8; // ncclUint64
+    __half*        F2; // ncclFloat16
    float*         F4; // ncclFloat32
    double*        F8; // ncclFloat64
    rccl_bfloat16* B2; // ncclBfloat16