Hey all, I’ve been trying to use Flux on my GPU and am running into a scalar indexing error when trying to multiply a matrix by a scalar. I found a workaround by creating a function that multiplies two items and then broadcasting the function mul_scalar(x,a)=x*a # using this in flux works as expec…

I think the problem here is that your scalar k=0.25 is Float64, while the arrays are Float32. This promotes the array (which you probably don’t want, for performance reasons), and that leads to mixed element types in the gradient, which is what’s giving an error: julia> using CUDA, LinearAlgebra j…

Scalar Indexing Error: multiplying matrix by scalar

Specific Domains GPU

naubertc October 25, 2021, 4:25pm 3

Yes, you are correct. Thank you for your help!

Topic		Replies	Views
Overcoming Slow Scalar Operations on GPU Arrays GPU performance	19	6367	January 4, 2021
Scalar indexing GPU problem in Flux.jl model GPU question , flux	4	407	May 8, 2024
What's wrong with this Flux model definitin? Machine Learning first-steps , flux	1	767	November 7, 2019
Zygote + CUDA: scalar getindex with custom activation function using multiplication GPU flux	3	736	September 3, 2020
Scalar indexing error when loading MNIST Machine Learning	2	596	March 8, 2022

Scalar Indexing Error: multiplying matrix by scalar

Related topics