Reduce max chunksize - Githubissues

LuxDL / Lux.jl

Elegant & Performant Scientific Machine Learning in Julia

https://lux.csail.mit.edu/

MIT License

456 stars 54 forks source link

Closed avik-pal closed 2 months ago

avik-pal commented 2 months ago

Fixes #636

I thought the compiler specializes till ntuple size 12, but that doesn't seem to be the case. Capping chunksize at 8.