From 20c24b68eb3929a0067bead7e30cfd9759ffb00d Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Tue, 7 Jan 2025 10:55:55 -0500
Subject: [PATCH] fix: missing scalar indexing check for setindex

---
 src/TracedRArray.jl | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/TracedRArray.jl b/src/TracedRArray.jl
index 8244bfc24..671e39363 100644
--- 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