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

IFC mix build (#998)

Browse Source
This commit is contained in:
Bertan Dogancay
2023-12-02 18:49:52 -07:00
committed by GitHub
parent 4ba65d1d6a
commit 7c0f49a878
13 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/include/collectives.h
+1 -1
View File
@@ -39,7 +39,7 @@ struct ncclDevRedOpFull {
nccl##func##algo##proto
/* Declare all collective operations */
#ifdef USE_INDIRECT_FUNCTION_CALL
#if defined(USE_INDIRECT_FUNCTION_CALL) && !defined(__gfx940__) && !defined(__gfx941__) && !defined(__gfx942__)
#define DECL5(func, algo, proto, devredop, type) \
extern __device__ void NCCL_FUNC_NAME(func, algo, proto, devredop, type)(); \
extern __global__ void NCCL_KERN_NAME(func, algo, proto, devredop, type)(struct ncclDevComm* comm, uint64_t channelMask, struct ncclWork* workHead); \