diff --git a/test/backend/test_schedule.py b/test/backend/test_schedule.py index 5693018899..804b4718c9 100644 --- a/test/backend/test_schedule.py +++ b/test/backend/test_schedule.py @@ -4,8 +4,6 @@ import unittest import numpy as np -from typing import cast -from hypothesis import assume, given, strategies as strat from tinygrad import nn, dtypes, Device, Tensor, Variable from tinygrad.uop.ops import UOp, Ops, UPat @@ -87,25 +85,6 @@ class TestSchedule(unittest.TestCase): x = Tensor.arange(768).reshape(3, 256).float() np.testing.assert_allclose((x.sum(axis=1) + x.sum(axis=1).sum()).numpy(), x.numpy().sum(axis=1) + x.numpy().sum(axis=1).sum()) - @unittest.skip("disabling subbuffer manually isn't supported anymore") - def test_bitcast_disable_subbufer(self): - x = cast(UOp, Tensor.empty(1, dtype=dtypes.float32).realize().uop) - a = x.alu(Ops.EXP2).cast(dtypes.int32, True, allow_buffer_view=False) - b = x.cast(dtypes.int32, True, allow_buffer_view=False) - b = a.alu(Ops.ADD, b) - check_schedule(b, 1) - - @given(strat.sampled_from(dtypes.all), strat.sampled_from(dtypes.all)) - @unittest.skip("kernel count depends on input") - def test_cast_padded_const(self, dt1, dt2): - assume(dt1 in supported_dtypes and dt2 in supported_dtypes) - a = Tensor(1, dtype=dt1).reshape(1, 1).pad(((1, 1), None)) - casted_view = a.cast(dt2) - run_linear(*check_schedule(casted_view, 0)) - realized_const_view = casted_view.contiguous() - run_linear(*check_schedule(realized_const_view, 1)) - np.testing.assert_equal(realized_const_view.numpy(), [[0], [1], [0]]) - def test_fuse_assign_contiguous(self): x = Tensor.zeros(4, 4, dtype=dtypes.int).contiguous().realize() a = Tensor.arange(8).reshape(4, 2)