Method test_sum

test/test_tiny.py:39–41

(self, N=getenv("SUM_N", 256))

    self.assertListEqual(out.tolist(), [1]*8+[0]*8)

  def test_sum(self, N=getenv("SUM_N", 256)):
    out = Tensor.ones(N).contiguous().sum()
    self.assertEqual(out.item(), N)

  def test_gemm(self, N=getenv("GEMM_N", 64)):
    a = Tensor.ones(N,N).contiguous()