polish

Author: Fe-r-oz

ext/QuantumToolboxCUDAExt.jl

@@ -5,7 +5,7 @@ using QuantumToolbox: makeVal, getVal
 import QuantumToolbox: _sparse_similar, _convert_eltype_wordsize
 import CUDA: cu, CuArray, allowscalar, @allowscalar, has_cuda
 import CUDA.CUSPARSE: CuSparseVector, CuSparseMatrixCSC, CuSparseMatrixCSR, AbstractCuSparseArray
-import SparseArrays: SparseVector, SparseMatrixCSC, sparse
+import SparseArrays: SparseVector, SparseMatrixCSC, sparse, spzeros
 import CUDA.Adapt: adapt
 
 export _safe_setindex!
@@ -112,4 +112,5 @@ _sparse_similar(A::CuSparseMatrixCSC, m::Int, n::Int) = CuSparseMatrixCSC(spzero
 _sparse_similar(A::CuSparseMatrixCSR, I::AbstractVector, J::AbstractVector, V::AbstractVector, m::Int, n::Int) =
     CuSparseMatrixCSR(sparse(I, J, V, m, n))
 _sparse_similar(A::CuSparseMatrixCSR, m::Int, n::Int) = CuSparseMatrixCSR(spzeros(eltype(A), m, n))
+
 end

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
+    allowed_setindex!(v0, weight, n_max*N+1)  # Safe GPU indexing scalar indexing is not allowed in GPU
 
     # Handle preconditioners
     (haskey(kwargs, :Pl) || haskey(kwargs, :Pr)) && error("The use of preconditioners must be defined in the solver.")