test: `BFloat16` testing

LuxDL / LuxLib.jl

Backend for Lux.jl

MIT License

9 stars 0 forks source link

Open avik-pal opened 3 months ago

avik-pal commented 3 months ago

first merge #114

[ ] Enzyme support? https://github.com/EnzymeAD/Enzyme.jl/issues/1695
[ ] Does AMDGPU support support BFloat16?
[ ] Our conv routines on CUDA will fail with BFloat16 -- do we write our own fallback implementations