diff options
author | Sylvain Jeaugey <sjeaugey@nvidia.com> | 2020-03-27 03:31:24 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-03-27 03:31:24 +0300 |
commit | 533e3702cf713a9ab9a634fbb8b4c380ecf381e6 (patch) | |
tree | 9e7eb6643f4356abd1a44e61de244b05db20c946 /src/collectives/device/reduce.h | |
parent | 6c61492eba5c25ac6ed1bf57de23c6a689aa75cc (diff) | |
parent | b221128ecacf4ce1b3054172b9f30163307042c5 (diff) |
Merge pull request #314 from NVIDIA/v2.6
2.6.4-1
Diffstat (limited to 'src/collectives/device/reduce.h')
-rw-r--r-- | src/collectives/device/reduce.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/collectives/device/reduce.h b/src/collectives/device/reduce.h index 0680abe..e36613f 100644 --- a/src/collectives/device/reduce.h +++ b/src/collectives/device/reduce.h @@ -50,6 +50,9 @@ __device__ void ncclReduceRingKernel(struct CollectiveArgs* args) { template<int UNROLL, class FUNC, typename T> __device__ void ncclReduceTreeKernel(struct CollectiveArgs* args) { } +template<int UNROLL, class FUNC, typename T> +__device__ void ncclReduceCollNetKernel(struct CollectiveArgs* args) { } + template<int UNUSED, class FUNC, typename T> __device__ void ncclReduceRingLLKernel(struct CollectiveArgs* args) { const int tid = threadIdx.x; @@ -94,6 +97,9 @@ __device__ void ncclReduceRingLLKernel(struct CollectiveArgs* args) { template<int UNUSED, class FUNC, typename T> __device__ void ncclReduceTreeLLKernel(struct CollectiveArgs* args) { } +template<int UNUSED, class FUNC, typename T> +__device__ void ncclReduceCollNetLLKernel(struct CollectiveArgs* args) { } + #include "prims_ll128.h" template<int UNUSED, class FUNC, typename T> __device__ void ncclReduceRingLL128Kernel(struct CollectiveArgs* args) { @@ -138,3 +144,6 @@ __device__ void ncclReduceRingLL128Kernel(struct CollectiveArgs* args) { template<int UNUSED, class FUNC, typename T> __device__ void ncclReduceTreeLL128Kernel(struct CollectiveArgs* args) { } + +template<int UNUSED, class FUNC, typename T> +__device__ void ncclReduceCollNetLL128Kernel(struct CollectiveArgs* args) { } |