7dfefbeb97
Fix NCCL build on clang with https://reviews.llvm.org/D68578. PiperOrigin-RevId: 352359401 Change-Id: I8990fa2dc7cc960c30a21073a7771c75d4b2678b
54 lines
2.1 KiB
Diff
54 lines
2.1 KiB
Diff
diff --git a/src/collectives/device/all_gather.cu b/src/collectives/device/all_gather.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/all_gather.cu
|
|
rename to src/collectives/device/all_gather.cu.cc
|
|
diff --git a/src/collectives/device/all_reduce.cu b/src/collectives/device/all_reduce.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/all_reduce.cu
|
|
rename to src/collectives/device/all_reduce.cu.cc
|
|
diff --git a/src/collectives/device/broadcast.cu b/src/collectives/device/broadcast.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/broadcast.cu
|
|
rename to src/collectives/device/broadcast.cu.cc
|
|
diff --git a/src/collectives/device/functions.cu b/src/collectives/device/functions.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/functions.cu
|
|
rename to src/collectives/device/functions.cu.cc
|
|
diff --git a/src/collectives/device/reduce.cu b/src/collectives/device/reduce.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/reduce.cu
|
|
rename to src/collectives/device/reduce.cu.cc
|
|
diff --git a/src/collectives/device/reduce_scatter.cu b/src/collectives/device/reduce_scatter.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/reduce_scatter.cu
|
|
rename to src/collectives/device/reduce_scatter.cu.cc
|
|
diff --git a/src/collectives/device/sendrecv.cu b/src/collectives/device/sendrecv.cu.cc
|
|
similarity index 100%
|
|
rename from src/collectives/device/sendrecv.cu
|
|
rename to src/collectives/device/sendrecv.cu.cc
|
|
diff --git a/src/nccl.h.in b/src/nccl.h
|
|
similarity index 98%
|
|
rename from src/nccl.h.in
|
|
rename to src/nccl.h
|
|
index 985274e..7ebb1e1 100644
|
|
--- a/src/nccl.h.in
|
|
+++ b/src/nccl.h
|
|
@@ -10,12 +10,12 @@
|
|
#include <cuda_runtime.h>
|
|
#include <cuda_fp16.h>
|
|
|
|
-#define NCCL_MAJOR ${nccl:Major}
|
|
-#define NCCL_MINOR ${nccl:Minor}
|
|
-#define NCCL_PATCH ${nccl:Patch}
|
|
-#define NCCL_SUFFIX "${nccl:Suffix}"
|
|
+#define NCCL_MAJOR 2
|
|
+#define NCCL_MINOR 8
|
|
+#define NCCL_PATCH 3
|
|
+#define NCCL_SUFFIX ""
|
|
|
|
-#define NCCL_VERSION_CODE ${nccl:Version}
|
|
+#define NCCL_VERSION_CODE 2803
|
|
#define NCCL_VERSION(X,Y,Z) ((X) * 1000 + (Y) * 100 + (Z))
|
|
|
|
#ifdef __cplusplus
|