issues
search
LuxDL
/
LuxLib.jl
Backend for Lux.jl
MIT License
9
stars
0
forks
source link
Integration of `oneDNN` for CPU operations
#150
Open
avik-pal
opened
2 months ago
avik-pal
commented
2 months ago
[x] Add oneDNN binaries to Yggdrasil --
https://github.com/JuliaPackaging/Yggdrasil/pull/9345
[ ] how expensive is it to construct the internal memory type for the operations?
[ ] create a
oneDNN.jl
lib that is framework agnostic. High level functionality along with integration with Enzyme and ChainRulesCore
[ ] convolution
[ ] batchnorm
[ ] groupnorm
[ ] fused_dense (low-priority)
[ ] integrate the primitives into LuxLib (via extensions)
[ ] convolution (highest priority)
[ ] activations (low-priority)
[ ] batchnorm
[ ] groupnorm
[ ] fused_dense (low-priority)
oneDNN.jl
lib that is framework agnostic. High level functionality along with integration with Enzyme and ChainRulesCore