From 28a67106ca3acb78ec83e545ac2e7eb4a4e2aea4 Mon Sep 17 00:00:00 2001 From: chenyu Date: Tue, 28 Nov 2023 01:33:41 -0500 Subject: [PATCH] enable symbolic ops tests for hip (#2485) --- .github/workflows/test.yml | 2 +- test/test_symbolic_ops.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 7ee17df231..586d8ed798 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -314,7 +314,7 @@ jobs: - name: Test HIP compilation on RDNA3 [gfx1100] run: | export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/hip/lib - MOCKHIP=1 HIP=1 python -m pytest -s test/test_hip_rdna3.py + MOCKHIP=1 HIP=1 python -m pytest -s test/test_hip_rdna3.py test/test_symbolic_ops.py tests: diff --git a/test/test_symbolic_ops.py b/test/test_symbolic_ops.py index 73036b7740..367e9daeaa 100644 --- a/test/test_symbolic_ops.py +++ b/test/test_symbolic_ops.py @@ -5,7 +5,7 @@ from tinygrad.tensor import Tensor, Device import numpy as np @unittest.skipIf(getenv("ARM64") or getenv("PTX"), "ARM64 and PTX are not supported") -@unittest.skipIf(Device.DEFAULT in ["HIP", "WEBGPU"], f"{Device.DEFAULT} is 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()