Add @const which does not impact performance

roflmaostc · Feb 11, 2024 · 00d8398 · 00d8398
1 parent bfa7919
commit 00d8398
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/backproject.jl b/src/backproject.jl
@@ -103,8 +103,8 @@ function _backproject(sinogram::AbstractArray{T, 3}, angles_T::AbstractVector, g
     return img 
 end
 
-@kernel function backproject_kernel2!(img::AbstractArray{T}, sinogram::AbstractArray{T}, weights, in_height,
-                                 out_height, angles, mid, radius, absorb_f) where {T}
+@kernel function backproject_kernel2!(img::AbstractArray{T}, @Const(sinogram), @Const(weights), @Const(in_height),
+        @Const(out_height), @Const(angles), mid, radius, absorb_f) where {T}
     i, iangle, i_z = @index(Global, NTuple)
 
     @inbounds sinα, cosα = sincos(angles[iangle])

diff --git a/src/radon.jl b/src/radon.jl
@@ -177,8 +177,8 @@ end
 
 @inline inside_circ(ii, jj, mid, radius) = (ii - mid)^2 + (jj - mid)^2 ≤ radius ^2 
 
-@kernel function radon_kernel!(sinogram::AbstractArray{T}, img::AbstractArray{T}, 
-                               weights, in_height, out_height, angles, mid,
+@kernel function radon_kernel!(sinogram::AbstractArray{T}, @Const(img), 
+                                @Const(weights), @Const(in_height), @Const(out_height), @Const(angles), mid,
                                radius, absorb_f) where {T}
     i, iangle, i_z = @index(Global, NTuple)