JuliaGPU/CUDA.jl

cuDNN: Provide wrappers for the declarative API

Open

#2266 aperta il 14 feb 2024

Vedi su GitHub
 (3 commenti) (0 reazioni) (0 assegnatari)Julia (274 fork)batch import
enhancementhelp wanted

Metriche repository

Star
 (1408 star)
Metriche merge PR
 (Merge medio 5g 5h) (16 PR mergiate in 30 g)

Descrizione

As part of cuDNN 9.0, the old imperative API has been deprecated, while moving towards a new, graph-based declarative API. That means that our wrappers will have to be rewritten. As I'm not familiar with machine learning / NNlib.jl (which I guess these wrappers would need to be compatible with), I'm hoping that people from the ML community could be able to help with this.

cc @ToucheSir @CarloLucibello @mcabbott @DhairyaLGandhi

Guida contributor