Merge remote-tracking branch 'nccl/master' into develop

2025-04-30 13:30:11 -05:00
parent dc0c5f9153 80f6bda437
commit cb6e23ae67
41 changed files with 1315 additions and 504 deletions
@@ -70,6 +70,9 @@ DECLARE_ROCM_PFN_EXTERN(hsa_status_string);
 extern int ncclCuMemEnable();
 extern int ncclCuMemHostEnable();

+// Handle type used for cuMemCreate()
+extern CUmemAllocationHandleType ncclCuMemHandleType;
+
 ncclResult_t rocmLibraryInit(void);

 extern bool ncclCudaLaunchBlocking; // initialized by ncclCudaLibraryInit()