Skip to content

GitLab

Unverified Commit 18921471 authored by Pruthvi Madugundu's avatar Pruthvi Madugundu Committed by GitHub
Browse files

Update rccl header include path (#110)

parent 7a428776
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
apex/contrib/csrc/nccl_p2p/nccl_p2p_cuda.cu
View file @ 18921471
......@@ -6,7 +6,7 @@
#include <ctime>
#include <cassert>
#ifdef __HIP_PLATFORM_HCC__
#include "rccl.h"
#include "rccl/rccl.h"
#else
#include "nccl.h"
#endif
......
apex/contrib/csrc/peer_memory/peer_memory_cuda.cu
View file @ 18921471
......@@ -8,7 +8,7 @@
#ifdef __HIP_PLATFORM_HCC__
#include <hip/hip_cooperative_groups.h>
#include "rccl.h"
#include "rccl/rccl.h"
#else
#include <cooperative_groups.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