saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Fix] USE_NCCL -> ORT_USE_NCCL (#21136)

Browse source 
### Description
Correct the macro used when NCCL enabled.
This commit is contained in:
mindest 2024-06-25 02:33:17 +08:00 committed by GitHub
parent 3a917e49fb
commit adaf0e8116
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/contrib_ops/cuda/collective/custom_reduce_impl.cu
View file

@ -29,7 +29,7 @@ namespace onnxruntime {
namespace cuda {
namespace collective {
#if defined(USE_MPI) || defined(USE_NCCL)
#if defined(USE_MPI) || defined(ORT_USE_NCCL)
using namespace onnxruntime;
using namespace onnxruntime::cuda;

2
onnxruntime/contrib_ops/cuda/collective/custom_reduce_impl.h
View file

@ -25,7 +25,7 @@ namespace onnxruntime {
namespace cuda {
namespace collective {
#if defined(USE_MPI) || defined(USE_NCCL)
#if defined(USE_MPI) || defined(ORT_USE_NCCL)
constexpr size_t WARP_SIZE = 32;
constexpr size_t MAX_ALL_REDUCE_BLOCKS = 24;

2
onnxruntime/contrib_ops/cuda/collective/ipc_utils.cc
View file

@ -23,7 +23,7 @@ namespace onnxruntime {
namespace cuda {
namespace collective {
#if defined(USE_MPI) || defined(USE_NCCL)
#if defined(USE_MPI) || defined(ORT_USE_NCCL)
using namespace onnxruntime;

4
onnxruntime/contrib_ops/cuda/collective/ipc_utils.h
View file

@ -24,7 +24,7 @@ namespace onnxruntime {
namespace cuda {
namespace collective {
#if defined(USE_MPI) || defined(USE_NCCL)
#if defined(USE_MPI) || defined(ORT_USE_NCCL)
struct CudaDeleter {
void operator()(void* ptr) const noexcept {
@ -86,4 +86,4 @@ GetCustomAllReduceWorkspace(int rank, int world_size, size_t input_size, IPCMemo
} // namespace collective
} // namespace cuda
} // namespace onnxruntime
} // namespace onnxruntime

2
onnxruntime/contrib_ops/cuda/collective/nccl_kernels.h
View file

@ -5,7 +5,7 @@
#include "core/providers/cuda/cuda_kernel.h"
#if defined(ORT_USE_NCCL) || defined(ORT_USE_MPI)
#if defined(ORT_USE_NCCL) || defined(USE_MPI)
#ifndef USE_ROCM
#include "custom_reduce_impl.h"
#include "ipc_utils.h"