JuliaGPU / JuliaGPU/NCCL.jl

Complex number wrapper

Open
#56 6 comments 0 reactions 0 assignees View on GitHub
Dominant language
Julia
Stars
31
Forks
15
PR merge metrics
No merged PRs in 30d

Description

NCCL does not support complex numbers directly and does not plan to ([see issue](https://github.com/NVIDIA/nccl/issues/539)). Are we willing to add a wrapper to NCCL.jl to make using complex numbers more convienient? Alternatively, the wrapper could be put in a higher level package (ex. Lux.jl, [see issue](https://github.com/LuxDL/Lux.jl/issues/747)). I am happy to start working on this but would like some feedback if possible. My primary motivation is using neural networks with complex valued weights and this feature would greatly simplify things.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.