Simplify kernel

JuliaGPU · Dec 16, 2024 · fc94e65 · fc94e65
1 parent 5bdfbed
commit fc94e65
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/host/linalg.jl b/src/host/linalg.jl
@@ -745,9 +745,7 @@ function LinearAlgebra.kron!(z::AbstractGPUVector{T1}, x::AbstractGPUVector{T2},
     @kernel function kron_kernel!(z, @Const(x), @Const(y))
         i, j = @index(Global, NTuple)
 
-        ly = length(y)
-
-        @inbounds z[(i - 1) * ly + j] = x[i] * y[j]
+        @inbounds z[(i - 1) * length(y) + j] = x[i] * y[j]
     end
 
     backend = KernelAbstractions.get_backend(z)