fix viz [pr] (#7519)

* fix viz [pr] * Update serve.py
2026-04-29 03:00:14 -04:00 · 2024-11-04 15:02:41 +08:00
parent 6bb230287b
commit 9a7cc04843
2 changed files with 5 additions and 1 deletions
--- a/test/test_tiny.py
+++ b/test/test_tiny.py
@@ -19,6 +19,10 @@ class TestTiny(unittest.TestCase):
    out = Tensor.cat(Tensor.ones(8).contiguous(), Tensor.ones(8).contiguous())
    self.assertListEqual(out.tolist(), [1]*16)

+  def test_sum(self):
+    out = Tensor.ones(256).contiguous().sum()
+    self.assertEqual(out.item(), 256)
+
  def test_gemm(self, N=4, out_dtype=dtypes.float):
    a = Tensor.ones(N,N).contiguous()
    b = Tensor.eye(N).contiguous()