|
55 | 55 | @test compare(A->mapreduce(x->x+x, +, A; dims=dims, init=zero(ET)), AT, rand(range, sz))
|
56 | 56 | end
|
57 | 57 | end
|
| 58 | + # Test more corner cases. Tests from AcceleraterKernels.jl |
| 59 | + for dims in [1,2,3,4,[1,2],[1,3],[1,4],[2,3],[2,4],[3,4],[1,2,3],[1,2,4],[1,3,4],[2,3,4],[1,2,3,4]] |
| 60 | + for isize in 0:3 |
| 61 | + for jsize in 0:3 |
| 62 | + for ksize in 0:3 |
| 63 | + @test compare(A->mapreduce(x->x+x, +, A; init=zero(Int32), dims), AT, rand(Int32(1):Int32(10), isize, jsize, ksize)) |
| 64 | + end |
| 65 | + end |
| 66 | + end |
| 67 | + end |
58 | 68 | end
|
59 | 69 |
|
60 | 70 | @testsuite "reductions/reduce" (AT, eltypes)->begin
|
|
68 | 78 | @test compare(A->reduce(*, A; dims=dims, init=one(ET)), AT, rand(range, sz))
|
69 | 79 | end
|
70 | 80 | end
|
| 81 | + # Test more corner cases. Tests from AcceleraterKernels.jl |
| 82 | + for dims in [1,2,3,4,[1,2],[1,3],[1,4],[2,3],[2,4],[3,4],[1,2,3],[1,2,4],[1,3,4],[2,3,4],[1,2,3,4]] |
| 83 | + for isize in 0:3 |
| 84 | + for jsize in 0:3 |
| 85 | + for ksize in 0:3 |
| 86 | + @test compare(A->reduce(+, A; init=zero(Int32), dims), AT, rand(Int32(1):Int32(10), isize, jsize, ksize)) |
| 87 | + end |
| 88 | + end |
| 89 | + end |
| 90 | + end |
71 | 91 | end
|
72 | 92 |
|
73 | 93 | @testsuite "reductions/sum prod" (AT, eltypes)->begin
|
|
0 commit comments