Improve cuda error (#480)

EnzymeAD · Jan 6, 2025 · 03b8363 · 03b8363
1 parent 80e4f28
commit 03b8363
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ext/ReactantCUDAExt.jl b/ext/ReactantCUDAExt.jl
@@ -353,7 +353,9 @@ Reactant.@reactant_overlay @noinline function (func::LLVMFunc{F,tt})(
     aliases = MLIR.IR.Attribute[]
     rarrays = TracedRArray[]
     for (i, a) in enumerate(args)
-        @assert a isa CuTracedArray
+        if !(a isa CuTracedArray)
+            throw(AssertionError("Expected a CuTracedArray found $(Core.Typeof(a)), $a"))
+        end
         ta =
             Base.unsafe_pointer_to_objref(Base.reinterpret(Ptr{Cvoid}, a.ptr))::TracedRArray
         push!(rarrays, ta)