From 03600aef1eda68f16e5abb62e189ff54b1dac01f Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Wed, 10 Dec 2025 22:03:06 -0500
Subject: [PATCH] failed test case when init jit with empty inputs (#13641)

not related to bert grad acc, but still seems to be a bug
---
 test/test_jit.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/test/test_jit.py b/test/test_jit.py
index d5c7ee383e..7b0b5dc5f4 100644
--- a/test/test_jit.py
+++ b/test/test_jit.py
@@ -501,6 +501,18 @@ class TestJit(unittest.TestCase):
     b = f(Tensor([2.0]))
     assert abs((a - b).item()) > 0.5
 
+  def test_jit_init_with_empty_different_size(self):
+    @TinyJit
+    def f(x:Tensor) -> Tensor: return (x + 1).realize()
+
+    f(Tensor.empty(1))
+    f(Tensor.empty(1))
+    # TODO: this should fail since input has a different size
+    f(Tensor(2.0)).item()
+    # TODO: this should not fail, and should return 3
+    with self.assertRaises(AssertionError):
+      f(Tensor([2.0])).item()
+
 @unittest.skip("Pending multioutput implementation #3607")
 class TestMultioutputJit(unittest.TestCase):
   def _test(self, f):