Skip to content

Commit 7cd524c

Browse files
Split tensor/test_blockwise.py to a separate CI job (#1820)
* Split blockwise and subtensor tests to deparate CI job * Skip dimshuffle memory leak test in numba CI
1 parent 3efb27e commit 7cd524c

File tree

3 files changed

+8
-3
lines changed

3 files changed

+8
-3
lines changed

.github/workflows/test.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,9 @@ jobs:
8484
part:
8585
- [ "*rest", "tests --ignore=tests/scan --ignore=tests/tensor --ignore=tests/xtensor --ignore=tests/link/numba" ]
8686
- [ "scan", "tests/scan" ]
87-
- [ "tensor *rest", "tests/tensor --ignore=tests/tensor/test_basic.py --ignore=tests/tensor/test_elemwise.py --ignore=tests/tensor/test_math.py --ignore=tests/tensor/test_math_scipy.py --ignore=tests/tensor/test_blas.py --ignore=tests/tensor/signal --ignore=tests/tensor/rewriting --ignore=tests/tensor/linalg --ignore=tests/tensor/test_nlinalg.py --ignore=tests/tensor/test_slinalg.py --ignore=tests/tensor/test_pad.py" ]
88-
- [ "tensor basic+elemwise", "tests/tensor/test_basic.py tests/tensor/test_elemwise.py" ]
89-
- [ "tensor math", "tests/tensor/test_math.py" ]
87+
- [ "tensor *rest", "tests/tensor --ignore=tests/tensor/test_basic.py --ignore=tests/tensor/test_elemwise.py --ignore=tests/tensor/test_math.py --ignore=tests/tensor/test_math_scipy.py --ignore=tests/tensor/test_blas.py --ignore=tests/tensor/signal --ignore=tests/tensor/rewriting --ignore=tests/tensor/linalg --ignore=tests/tensor/test_nlinalg.py --ignore=tests/tensor/test_slinalg.py --ignore=tests/tensor/test_pad.py --ignore tests/tensor/test_blockwise.py --ignore tests/tensor/test_subtensor.py" ]
88+
- [ "tensor basic+elemwise+blockwise", "tests/tensor/test_basic.py tests/tensor/test_elemwise.py tests/tensor/test_blockwise.py " ]
89+
- [ "tensor math+subtensor", "tests/tensor/test_math.py tests/tensor/test_subtensor.py" ]
9090
- [ "tensor scipy+blas+pad", "tests/tensor/test_math_scipy.py tests/tensor/test_blas.py tests/tensor/signal tests/tensor/test_pad.py" ]
9191
- [ "tensor rewriting", "tests/tensor/rewriting" ]
9292
- [ "tensor linalg", "tests/tensor/linalg tests/tensor/test_nlinalg.py tests/tensor/test_slinalg.py" ]

tests/tensor/test_einsum.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
"error",
2222
r"ignore:^Numba will use object mode to run.*perform method\.:UserWarning",
2323
r"ignore:Cannot cache compiled function \"numba_funcified_fgraph\".*:numba.NumbaWarning",
24+
r"ignore::numba.NumbaPerformanceWarning",
2425
)
2526

2627
floatX = pytensor.config.floatX

tests/tensor/test_elemwise.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,6 +210,10 @@ def test_c_views(self):
210210
# Confirm the broadcasted value in the output
211211
assert np.array_equiv(outputs[0].storage[0], 2039)
212212

213+
@pytest.mark.skipif(
214+
isinstance(get_default_mode().linker, NumbaLinker),
215+
reason="Running with numba linker, c backend (should be) covered in another CI",
216+
)
213217
@pytest.mark.parametrize("inplace", [True, False])
214218
def test_memory_leak(self, inplace):
215219
import gc

0 commit comments

Comments
 (0)