commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hotfix: don't run OOM test in CI

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
Show Stats 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 import unittest
from tinygrad import Device from tinygrad import Device
from tinygrad.helpers import CI
from tinygrad.runtime.ops_gpu import CLDevice, CLAllocator 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): class TestOCLOOM(unittest.TestCase):
def test_opencl_oom(self): def test_opencl_oom(self):
with self.assertRaises(RuntimeError) as err: with self.assertRaises(RuntimeError) as err: