polish

Author: Fe-r-oz

src/steadystate.jl

@@ -403,7 +403,7 @@ function _steadystate_fourier(
     # Create initial vector with GPU support
     v0 = _dense_similar(L_0_mat, n_fourier * N)
     fill!(v0, zero(T))
-    allowed_setindex!(v0, weight, n_max*N+1)  # Safe GPU indexing scalar indexing is not allowed in GPU
+    v0[n_max * N + 1 : n_max * N + 1] .= weight
 
     # Handle preconditioners
     (haskey(kwargs, :Pl) || haskey(kwargs, :Pr)) && error("The use of preconditioners must be defined in the solver.")