# Flux model with \`SubArray\` parameters don't work on GPU

**URL:** <https://discourse.julialang.org/t/flux-model-with-subarray-parameters-dont-work-on-gpu/60401>\
**Category:** GPU\
**Tags:** question, cuda, flux, zygote\
**Created:** [May 1, 2021, 9:25pm UTC](https://discourse.julialang.org/t/flux-model-with-subarray-parameters-dont-work-on-gpu/60401 "2021-05-01T21:25:37Z")\
**Posts on this page:** 1\
**Showing post:** 4

<div class="post-metadata">

**Author:** ![ArthurW](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/arthurw/32/19561_2.png) [@ArthurW](https://discourse.julialang.org/u/ArthurW)\
**Post date:** [August 31, 2021, 12:00pm UTC](https://discourse.julialang.org/t/flux-model-with-subarray-parameters-dont-work-on-gpu/60401/4 "2021-08-31T12:00:14Z")

</div>

It works now. I just tried it.

Many of the packages involved have been updated since I created the topic (4 months ago). Most notably, _CUDA.jl_ went from version _3.1.0_ to _3.3.4_.  
I haven’t touched the project in the meanwhile, so I don’t know which update solved the issue.

Thank you all for the help.

---

_[View the full topic](https://discourse.julialang.org/t/flux-model-with-subarray-parameters-dont-work-on-gpu/60401)._