Unverified 提交 7cd524ca authored 作者: Jesse Grabowski's avatar Jesse Grabowski 提交者: GitHub

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
上级 3efb27eb
...@@ -84,9 +84,9 @@ jobs: ...@@ -84,9 +84,9 @@ jobs:
part: part:
- [ "*rest", "tests --ignore=tests/scan --ignore=tests/tensor --ignore=tests/xtensor --ignore=tests/link/numba" ] - [ "*rest", "tests --ignore=tests/scan --ignore=tests/tensor --ignore=tests/xtensor --ignore=tests/link/numba" ]
- [ "scan", "tests/scan" ] - [ "scan", "tests/scan" ]
- [ "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" ] - [ "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" ]
- [ "tensor basic+elemwise", "tests/tensor/test_basic.py tests/tensor/test_elemwise.py" ] - [ "tensor basic+elemwise+blockwise", "tests/tensor/test_basic.py tests/tensor/test_elemwise.py tests/tensor/test_blockwise.py " ]
- [ "tensor math", "tests/tensor/test_math.py" ] - [ "tensor math+subtensor", "tests/tensor/test_math.py tests/tensor/test_subtensor.py" ]
- [ "tensor scipy+blas+pad", "tests/tensor/test_math_scipy.py tests/tensor/test_blas.py tests/tensor/signal tests/tensor/test_pad.py" ] - [ "tensor scipy+blas+pad", "tests/tensor/test_math_scipy.py tests/tensor/test_blas.py tests/tensor/signal tests/tensor/test_pad.py" ]
- [ "tensor rewriting", "tests/tensor/rewriting" ] - [ "tensor rewriting", "tests/tensor/rewriting" ]
- [ "tensor linalg", "tests/tensor/linalg tests/tensor/test_nlinalg.py tests/tensor/test_slinalg.py" ] - [ "tensor linalg", "tests/tensor/linalg tests/tensor/test_nlinalg.py tests/tensor/test_slinalg.py" ]
......
...@@ -21,6 +21,7 @@ pytestmark = pytest.mark.filterwarnings( ...@@ -21,6 +21,7 @@ pytestmark = pytest.mark.filterwarnings(
"error", "error",
r"ignore:^Numba will use object mode to run.*perform method\.:UserWarning", r"ignore:^Numba will use object mode to run.*perform method\.:UserWarning",
r"ignore:Cannot cache compiled function \"numba_funcified_fgraph\".*:numba.NumbaWarning", r"ignore:Cannot cache compiled function \"numba_funcified_fgraph\".*:numba.NumbaWarning",
r"ignore::numba.NumbaPerformanceWarning",
) )
floatX = pytensor.config.floatX floatX = pytensor.config.floatX
......
...@@ -210,6 +210,10 @@ class TestDimShuffle(unittest_tools.InferShapeTester): ...@@ -210,6 +210,10 @@ class TestDimShuffle(unittest_tools.InferShapeTester):
# Confirm the broadcasted value in the output # Confirm the broadcasted value in the output
assert np.array_equiv(outputs[0].storage[0], 2039) assert np.array_equiv(outputs[0].storage[0], 2039)
@pytest.mark.skipif(
isinstance(get_default_mode().linker, NumbaLinker),
reason="Running with numba linker, c backend (should be) covered in another CI",
)
@pytest.mark.parametrize("inplace", [True, False]) @pytest.mark.parametrize("inplace", [True, False])
def test_memory_leak(self, inplace): def test_memory_leak(self, inplace):
import gc import gc
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论