mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-04-29 03:00:14 -04:00
move fuzz_shape_ops to run with other fuzzer (#10767)
* move fuzz_shape_ops to run with other fuzzer * don't skip CPU
This commit is contained in:
2
.github/workflows/test.yml
vendored
2
.github/workflows/test.yml
vendored
@@ -394,6 +394,8 @@ jobs:
|
||||
run: |
|
||||
PYTHONPATH="." python test/external/fuzz_shapetracker.py
|
||||
PYTHONPATH="." python test/external/fuzz_shapetracker_math.py
|
||||
- name: Fuzz Test shape ops
|
||||
run: python test/external/fuzz_shape_ops.py
|
||||
|
||||
testgpuimage:
|
||||
name: 'GPU IMAGE Tests'
|
||||
|
||||
@@ -7,7 +7,7 @@ from hypothesis.extra import numpy as stn
|
||||
|
||||
import numpy as np
|
||||
import torch
|
||||
from tinygrad import Tensor, Device
|
||||
from tinygrad import Tensor
|
||||
from tinygrad.helpers import CI, getenv
|
||||
|
||||
|
||||
@@ -38,7 +38,6 @@ def apply(tor, ten, tor_fn, ten_fn=None):
|
||||
except: ten, ok = None, not ok # noqa: E722
|
||||
return tor, ten, ok
|
||||
|
||||
@unittest.skipIf(CI and Device.DEFAULT in ("CPU", "NV"), "slow")
|
||||
class TestShapeOps(unittest.TestCase):
|
||||
@settings.get_profile(__file__)
|
||||
@given(st_shape(), st_int32, st.one_of(st_int32, st.lists(st_int32)))
|
||||
Reference in New Issue
Block a user