Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[nccl] could we bring nvidia nccl to javacpp pytorch? #1402

Open
mullerhai opened this issue Aug 15, 2023 · 3 comments
Open

[nccl] could we bring nvidia nccl to javacpp pytorch? #1402

mullerhai opened this issue Aug 15, 2023 · 3 comments

Comments

@mullerhai
Copy link

HI, also for pytorch distribute training and infer use multiple Nviida Gpu communite message framework, https://github.com/NVIDIA/nccl ,eagerly need it code with javacpp-pytorch,thanks

@saudet
Copy link
Member

saudet commented Aug 15, 2023

I'm pretty sure that's already enabled. Could you explain the issue you're having with it?

@mullerhai
Copy link
Author

I'm pretty sure that's already enabled. Could you explain the issue you're having with it?

really,could you show me the nccl implement code in javacpp ? , and how about Nvidia apex https://github.com/NVIDIA/apex , has we already implement in javacpp

@mullerhai
Copy link
Author

mullerhai commented Aug 15, 2023

about nccl ,I only find the code location is here on cuda project https://github.com/bytedeco/javacpp-presets/blob/master/cuda/src/gen/java/org/bytedeco/cuda/nccl/... ,but only three little bit java file , why so small and https://github.com/bytedeco/javacpp-presets/blob/master/cuda/src/gen/java/org/bytedeco/cuda/global/nccl.java ,is all c++ version nccl function has implement ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants