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
Scalar indexing GPU problem in Flux.jl model GPU question , flux	4	436	May 8, 2024
Overcoming Slow Scalar Operations on GPU Arrays GPU performance	19	6450	January 4, 2021
Scalar indexing error when loading MNIST Machine Learning	2	604	March 8, 2022
Zygote + CUDA: scalar getindex with custom activation function using multiplication GPU flux	3	738	September 3, 2020
What's wrong with this Flux model definitin? Machine Learning first-steps , flux	1	770	November 7, 2019

Scalar Indexing Error: multiplying matrix by scalar

Related topics