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:
chenyu
2025-06-10 14:43:04 -07:00
committed by GitHub
parent 5e7ad70aae
commit 612cdf5146
2 changed files with 3 additions and 2 deletions

View File

@@ -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'

View File

@@ -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)))