Skip to content

Commit ee58829

Browse files
authored
Merge pull request #2676 from JuliaGPU/ksh/kernels
Exclude more device side code in CUSPARSE
2 parents 63de965 + 4ab0a9e commit ee58829

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

lib/cusparse/linalg.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,7 @@ function LinearAlgebra.dot(y::CuVector{T}, A::CuSparseMatrixCSC{T}, x::CuVector{
138138
end
139139
n = size(A, 2)
140140

141+
## COV_EXCL_START
141142
function kernel(y::CuDeviceVector{T1}, colPtr::CuDeviceVector{T2}, rowVal::CuDeviceVector{T2},
142143
nzVal::CuDeviceVector{T1}, x::CuDeviceVector{T1}, result::CuDeviceVector{T1}, n::Integer, shuffle) where {T1,T2}
143144

@@ -163,6 +164,7 @@ function LinearAlgebra.dot(y::CuVector{T}, A::CuSparseMatrixCSC{T}, x::CuVector{
163164
end
164165
return
165166
end
167+
## COV_EXCL_STOP
166168

167169
function compute_threads(max_threads, wanted_threads, shuffle, dev)
168170
if wanted_threads > max_threads
@@ -191,6 +193,7 @@ function LinearAlgebra.dot(y::CuVector{T}, A::CuSparseMatrixCSR{T}, x::CuVector{
191193
end
192194
n = size(A, 1)
193195

196+
## COV_EXCL_START
194197
function kernel(y::CuDeviceVector{T1}, rowPtr::CuDeviceVector{T2}, colVal::CuDeviceVector{T2},
195198
nzVal::CuDeviceVector{T1}, x::CuDeviceVector{T1}, result::CuDeviceVector{T1}, n::Integer, shuffle) where {T1,T2}
196199

@@ -216,6 +219,7 @@ function LinearAlgebra.dot(y::CuVector{T}, A::CuSparseMatrixCSR{T}, x::CuVector{
216219
end
217220
return
218221
end
222+
## COV_EXCL_STOP
219223

220224
function compute_threads(max_threads, wanted_threads, shuffle, dev)
221225
if wanted_threads > max_threads

0 commit comments

Comments
 (0)