Skip to content

apark263/nccl_wrapper

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NCCL wrapper for pycuda

We are attempting to create a shared library interface to NCCL primitives that would be usable by pycuda through ctypes.

Requirements

  • nccl.
  • PyCuda: In order for the cuda driver level allocations to work with the runtime api calls in nccl, we need to be able to access primary contexts. Use this pycuda fork.
  • CUDA-7.0 or greater

About

ctypes wrapper for nccl with pycuda

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published