github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

cleanup tests Device[Device.DEFAULT] is always Compiled (#3645)

Browse Source
This commit is contained in:
chenyu
2024-03-07 11:15:42 -05:00
committed by GitHub
parent bdd62c7fd8
commit 906cc3a69b
7 changed files with 5 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_method_cache.py
View File

@@ -1,10 +1,8 @@
import unittest
from tinygrad import Tensor, Device, Variable
from tinygrad.device import Compiled
from examples.gpt2 import Transformer
from tinygrad.nn.state import get_state_dict
@unittest.skipIf(not isinstance(Device[Device.DEFAULT], Compiled), "only test for compiled backends")
class TestMethodCache(unittest.TestCase):
def setUp(self):
self.backup_compiler = Device[Device.DEFAULT].compiler