Bugfix.

JuliaGPU · Oct 31, 2023 · 2d64194 · 2d64194
1 parent 82e6365
commit 2d64194
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/host/indexing.jl b/src/host/indexing.jl
@@ -154,7 +154,7 @@ function gpu_checkindex_indices(A, IA::Tuple, I::Tuple)
 end
 function gpu_checkindex_indices(A, ::Tuple{}, I::Tuple)
     @inline
-    gpu_checkindex(A, OneTo(1), I[1])::Bool & gpu_checkindex_indices(A, (), tail(I))
+    gpu_checkindex(A, Base.OneTo(1), I[1])::Bool & gpu_checkindex_indices(A, (), tail(I))
 end
 gpu_checkindex_indices(A, IA::Tuple, ::Tuple{}) = (@inline; all(x->length(x)==1, IA))
 gpu_checkindex_indices(A, ::Tuple{}, ::Tuple{}) = true

diff --git a/test/testsuite/indexing.jl b/test/testsuite/indexing.jl
@@ -129,6 +129,12 @@ end
         @test_throws DimensionMismatch x[1:9,1:9,:,:] = y
     end
 
+    @testset "mismatching axes/indices" begin
+        a = rand(1,1)
+        @test compare(a->a[1:1], AT, a)
+        @test compare(a->a[1:1,1:1], AT, a)
+        @test compare(a->a[1:1,1:1,1:1], AT, a)
+    end
 end
 
 @testsuite "indexing find" (AT, eltypes)->begin