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

support vals on WebGPU, run more tests (#2668)

Browse Source 
* Vals on webgpu, run more tests

* Skip slow tests, run symbolic ops tests

* Balance out tests
This commit is contained in:
Ahmed Harmouche
2023-12-08 01:45:21 +01:00
committed by GitHub
parent d02ff21f1a
commit 4b01839774
8 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_symbolic_ops.py
View File

@@ -1,11 +1,10 @@
import unittest
from tinygrad.shape.symbolic import Variable
from tinygrad.helpers import getenv
from tinygrad.tensor import Tensor, Device
from tinygrad.tensor import Tensor
import numpy as np
@unittest.skipIf(getenv("ARM64") or getenv("PTX"), "ARM64 and PTX are not supported")
@unittest.skipIf(Device.DEFAULT in ["WEBGPU"], f"{Device.DEFAULT} is not supported")
class TestSymbolicOps(unittest.TestCase):
def test_plus1(self):
def f(a): return (a+1).realize()