Skip to content

GitLab

Commit 2d8b3600 authored by Pruthvi Madugundu's avatar Pruthvi Madugundu Committed by flyingdown
Browse files

Update rccl header include path (#110)

parent e519c1e3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
apex/contrib/csrc/nccl_p2p/nccl_p2p_cuda.cu
View file @ 2d8b3600
...@@ -6,7 +6,7 @@ ...@@ -6,7 +6,7 @@
#include <ctime> #include <ctime>
#include <cassert> #include <cassert>
#ifdef __HIP_PLATFORM_HCC__ #ifdef __HIP_PLATFORM_HCC__
#include "rccl.h" #include "rccl/rccl.h"
#else #else
#include "nccl.h" #include "nccl.h"
#endif #endif
......
apex/contrib/csrc/peer_memory/peer_memory_cuda.cu
View file @ 2d8b3600
...@@ -8,7 +8,7 @@ ...@@ -8,7 +8,7 @@
#ifdef __HIP_PLATFORM_HCC__ #ifdef __HIP_PLATFORM_HCC__
#include <hip/hip_cooperative_groups.h> #include <hip/hip_cooperative_groups.h>
#include "rccl.h" #include "rccl/rccl.h"
#else #else
#include <cooperative_groups.h> #include <cooperative_groups.h>
#include "nccl.h" #include "nccl.h"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment