fix: missing scalar indexing check for setindex

EnzymeAD · Jan 7, 2025 · 20c24b6 · 20c24b6
1 parent 46b8c14
commit 20c24b6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/TracedRArray.jl b/src/TracedRArray.jl
@@ -175,7 +175,8 @@ end
 function maybe_assert_scalar_setindexing(
     ::TracedRArray{T,N}, ::Vararg{Union{Int,TracedRNumber{Int}},N}
 ) where {T,N}
-    return GPUArraysCore.assertscalar("setindex!(::TracedRArray, v, ::Vararg{Int, N})")
+    GPUArraysCore.assertscalar("setindex!(::TracedRArray, v, ::Vararg{Int, N})")
+    return
 end
 
 maybe_assert_scalar_setindexing(args...) = nothing