From 5d6e1df22596ef6b6a74de09f8344c15ec0cb185 Mon Sep 17 00:00:00 2001
From: Sophie L <sophie.lequeu@student.uclouvain.be>
Date: Thu, 2 Oct 2025 12:47:46 +0200
Subject: [PATCH] Check scalar output

---
 src/reverse_mode.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/reverse_mode.jl b/src/reverse_mode.jl
index baab908..fff3d74 100644
--- a/src/reverse_mode.jl
+++ b/src/reverse_mode.jl
@@ -323,6 +323,7 @@ function _forward_eval(
             f.partials_storage[rhs] = zero(T)
         end
     end
+    @assert f.sizes.ndims[1] == 0 "Final result must be scalar, got ndims = $(f.sizes.ndims[1])"
     return f.forward_storage[1]
 end