Skip to content

Commit 846c77d

Browse files
committed
remove scalar indexing
1 parent 3b8daa9 commit 846c77d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/gpu_all.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,12 @@ using LinearAlgebra, OrdinaryDiffEq, Test, PreallocationTools, CUDA, ForwardDiff
44
chunk_size = 5
55
u0_CU = cu(ones(5,5))
66
dual_CU = cu(zeros(ForwardDiff.Dual{ForwardDiff.Tag{typeof(something), Float32}, Float32, chunk_size}, 2, 2))
7+
dual_N = ForwardDiff.Dual{ForwardDiff.Tag{typeof(something), Float32}, Float32, 5}(0)
78
cache_CU = dualcache(u0_CU, chunk_size)
89
tmp_du_CUA = get_tmp(cache_CU, u0_CU)
910
tmp_dual_du_CUA = get_tmp(cache_CU, dual_CU)
10-
tmp_du_CUN = get_tmp(cache_CU, u0_CU[1])
11-
tmp_dual_du_CUN = get_tmp(cache_CU, dual_CU[1])
11+
tmp_du_CUN = get_tmp(cache_CU, 0.0)
12+
tmp_dual_du_CUN = get_tmp(cache_CU, dual_N)
1213
@test typeof(cache_CU.dual_du) == typeof(u0_CU) #check that dual cache array is a GPU array for performance reasons.
1314
@test size(tmp_du_CUA) == size(u0_CU)
1415
@test typeof(tmp_du_CUA) == typeof(u0_CU)

0 commit comments

Comments
 (0)