-
-
Notifications
You must be signed in to change notification settings - Fork 35
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Scalar indexing of a GPU array error #150
Comments
If you have moved the array to the CPU how could it complain about scalar indexing? Please put together a MWE |
@jonniedie any reason the CRC rrule for getproperty was added. It breaks compatibility with CuArrays. See LuxDL/Lux.jl#92 |
The last update is for fixing #148 I opened 🥲. A few hours ago I was celebrating the fact that it got fixed. Maybe we can try to figure out how to fix it while avoiding scalar indexing |
The PR #151 will fix both. |
Awesome. Thanks! |
Zygote complains with a
scalar indexing is disallowed error
with #999c335.My code does work without that change. Sorry, I don't have a MWE to share at the moment.
Maybe the problem could be that during the FW pass of my custom Lux model I move an array to the CPU before I use it to solve an ODE and then the solution back to the GPU?
In case it's impossible to tell what could cause the error, I can try to reduce my model to a MWE.
Feel free to close this issue if you think it's a duplicate of #126
The text was updated successfully, but these errors were encountered: