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

hotfix: don't run OOM test in CI

Browse Source
This commit is contained in:
George Hotz
2024-08-07 22:19:29 -07:00
parent 859d0e4709
commit c5baa3d66b
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_ocl.py
View File

@@ -1,8 +1,9 @@
import unittest
from tinygrad import Device
from tinygrad.helpers import CI
from tinygrad.runtime.ops_gpu import CLDevice, CLAllocator
@unittest.skipUnless(Device.DEFAULT in ["GPU"], "Runs only on OpenCL (GPU)")
@unittest.skipUnless(Device.DEFAULT in ["GPU"] and not CI, "Runs only on OpenCL (GPU)")
class TestOCLOOM(unittest.TestCase):
def test_opencl_oom(self):
with self.assertRaises(RuntimeError) as err: