Skip to content

Commit 47f4ea1

Browse files
committed
minor opt
1 parent fcc63e4 commit 47f4ea1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/host/reverse.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ function _reverse(input::AnyGPUArray{T, N}, output::AnyGPUArray{T, N};
1717

1818
## COV_EXCL_START
1919
@kernel unsafe_indices=true function kernel(input, output)
20-
offset_in = @groupsize()[1] * (@index(Group, Linear) - 1i32)
20+
offset_in = Int32(@groupsize()[1]) * (@index(Group, Linear) - 1i32)
2121
index_in = offset_in + @index(Local, Linear)
2222

2323
@inbounds if index_in <= length(input)
@@ -52,7 +52,7 @@ function _reverse!(data::AnyGPUArray{T, N}; dims=1:ndims(data)) where {T, N}
5252

5353
## COV_EXCL_START
5454
@kernel unsafe_indices=true function kernel(data)
55-
offset_in = @groupsize()[1] * (@index(Group, Linear) - 1i32)
55+
offset_in = Int32(@groupsize()[1]) * (@index(Group, Linear) - 1i32)
5656
index_in = offset_in + @index(Local, Linear)
5757

5858
@inbounds if index_in <= reduced_length

0 commit comments

Comments
 (0)