2.7.3-1

Add support for A100 GPU and related platforms. Add support for CUDA 11. Add support for send/receive operations (beta). [ROCm/rccl commit: 5949d96f36]
2020-05-12 14:40:18 -07:00
parent c43022b9d8
commit 89072c82e5
54 changed files with 2051 additions and 1254 deletions
@@ -0,0 +1,32 @@
+/*************************************************************************
+ * Copyright (c) 2015-2020, NVIDIA CORPORATION. All rights reserved.
+ *
+ * See LICENSE.txt for license information
+ ************************************************************************/
+
+#include <stdlib.h>
+
+#ifndef NCCL_P2P_H_
+#define NCCL_P2P_H_
+
+struct ncclP2Pinfo {
+ const void* sendbuff;
+  void* recvbuff;
+  ssize_t sendbytes;
+  ssize_t recvbytes;
+};
+
+struct ncclP2PConnect {
+  int nrecv[MAXCHANNELS];
+  int nsend[MAXCHANNELS];
+  int* recv;
+  int* send;
+};
+
+struct ncclP2Plist {
+  struct ncclP2Pinfo *peerlist;
+  int count;
+  struct ncclP2PConnect connect;
+};
+
+#endif