mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-07 21:26:21 -05:00
8798d120bb
* autopad shapetracker for BEAM * OptOps.PADTO * skip that test for now * correct padding reduce axis * just 32 * avoid more than double the FLOPs * cleanups * test case * no support for triton and llvm yet * typos * symbolic shape would not work * cannot PADTO with MAX kernel * advance db version * no breaking change - don't advance db version * is triton just python? * Revert "is triton just python?" This reverts commit 17e776c25587615e33a3634c2fb0bb8591ce65d4. * Revert "Revert "is triton just python?"" This reverts commit 6c434c01e1c4b0ea0431ec18632cd859fb3cf260. * support llvm * is it really passing in CI only? * update tests * oh triton test passed * simpler * revert that, with a test * check if st are the same * Revert "check if st are the same" This reverts commit d2a5eac110a5da1af82a2728c883779ef69c3cad. * update the db version * rebase artifact
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import unittest
|
|
import numpy as np
|
|
|
|
from tinygrad.helpers import BEAM, Timing
|
|
from tinygrad.shape.symbolic import Variable
|
|
from tinygrad.tensor import Tensor
|
|
from tinygrad.nn import Conv2d
|
|
|
|
class TestBeamSearch(unittest.TestCase):
|
|
def setUp(self):
|
|
self.old_beam = BEAM.value
|
|
BEAM.value = 2
|
|
def tearDown(self):
|
|
BEAM.value = self.old_beam
|
|
|
|
def test_variable_ast_beam(self):
|
|
a = Tensor.rand(3, 3).reshape((Variable("a", 1, 10).bind(3), 3))
|
|
a = (a+1).realize()
|
|
|
|
def test_big_prime_number(self):
|
|
a = Tensor.rand(367, 367)
|
|
b = Tensor.rand(367, 367)
|
|
c = (a@b).realize()
|
|
np.testing.assert_allclose(c.numpy(), a.numpy() @ b.numpy(), atol=1e-4, rtol=1e-4)
|
|
|
|
def test_variable_big_prime_number(self):
|
|
v = Variable("v", 1, 400).bind(367)
|
|
a = Tensor.rand(367, 367)
|
|
b = Tensor.rand(367, 367)
|
|
c = (a.reshape(367, v) @ b.reshape(v, 367)).realize()
|
|
np.testing.assert_allclose(c.numpy(), a.numpy() @ b.numpy(), atol=1e-4, rtol=1e-4)
|
|
|
|
def test_variable_shrink_prime_number(self):
|
|
v = Variable("v", 1, 400).bind(367)
|
|
a = Tensor.rand(400, 367)
|
|
b = (a.shrink(((0,v), None))+1).reshape(367,367).realize()
|
|
np.testing.assert_allclose(b.numpy(), a.numpy()[:367]+1, atol=1e-4, rtol=1e-4)
|
|
|
|
def test_no_mutate_rawbuffers(self):
|
|
a = Tensor.rand(3, 3).realize()
|
|
desired = a.numpy() + 1
|
|
a.assign(a+1)
|
|
actual = a.numpy()
|
|
np.testing.assert_allclose(actual, desired)
|
|
|
|
def test_conv_beam(self):
|
|
c = Conv2d(3, 16, (3,3))
|
|
x = Tensor.rand(1,3,32,32)
|
|
with Timing():
|
|
c(x).realize()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|