提交 43b3f401 authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Skip slow C CAReduce tests in numba linker CI

上级 7ede010a
...@@ -13,12 +13,13 @@ import pytensor.tensor as pt ...@@ -13,12 +13,13 @@ import pytensor.tensor as pt
import tests.unittest_tools as utt import tests.unittest_tools as utt
from pytensor import In, Out, config, grad from pytensor import In, Out, config, grad
from pytensor.compile.function import function from pytensor.compile.function import function
from pytensor.compile.mode import Mode from pytensor.compile.mode import Mode, get_default_mode
from pytensor.graph.basic import Apply, Variable from pytensor.graph.basic import Apply, Variable
from pytensor.graph.fg import FunctionGraph from pytensor.graph.fg import FunctionGraph
from pytensor.graph.replace import vectorize_node from pytensor.graph.replace import vectorize_node
from pytensor.link.basic import PerformLinker from pytensor.link.basic import PerformLinker
from pytensor.link.c.basic import CLinker, OpWiseCLinker from pytensor.link.c.basic import CLinker, OpWiseCLinker
from pytensor.link.numba import NumbaLinker
from pytensor.scalar import ScalarOp, float32, float64, int32, int64 from pytensor.scalar import ScalarOp, float32, float64, int32, int64
from pytensor.scalar import add as scalar_add from pytensor.scalar import add as scalar_add
from pytensor.scalar import exp as scalar_exp from pytensor.scalar import exp as scalar_exp
...@@ -658,6 +659,10 @@ class TestCAReduce(unittest_tools.InferShapeTester): ...@@ -658,6 +659,10 @@ class TestCAReduce(unittest_tools.InferShapeTester):
not pytensor.config.cxx, not pytensor.config.cxx,
reason="G++ not available, so we need to skip this test.", reason="G++ not available, so we need to skip this test.",
) )
@pytest.mark.skipif(
isinstance(get_default_mode().linker, NumbaLinker),
reason="Running with numba linker, c backend (should be) covered in another CI",
)
def test_c_noopt(self): def test_c_noopt(self):
# We need to make sure that we cover the corner cases that # We need to make sure that we cover the corner cases that
# optimizations normally cover # optimizations normally cover
...@@ -668,6 +673,10 @@ class TestCAReduce(unittest_tools.InferShapeTester): ...@@ -668,6 +673,10 @@ class TestCAReduce(unittest_tools.InferShapeTester):
not pytensor.config.cxx, not pytensor.config.cxx,
reason="G++ not available, so we need to skip this test.", reason="G++ not available, so we need to skip this test.",
) )
@pytest.mark.skipif(
isinstance(get_default_mode().linker, NumbaLinker),
reason="Running with numba linker, c backend (should be) covered in another CI",
)
def test_c(self): def test_c(self):
for dtype in ["bool", "floatX", "complex64", "complex128", "int8", "uint8"]: for dtype in ["bool", "floatX", "complex64", "complex128", "int8", "uint8"]:
self.with_mode(Mode(linker="c"), ps.add, dtype=dtype) self.with_mode(Mode(linker="c"), ps.add, dtype=dtype)
...@@ -687,6 +696,10 @@ class TestCAReduce(unittest_tools.InferShapeTester): ...@@ -687,6 +696,10 @@ class TestCAReduce(unittest_tools.InferShapeTester):
not pytensor.config.cxx, not pytensor.config.cxx,
reason="G++ not available, so we need to skip this test.", reason="G++ not available, so we need to skip this test.",
) )
@pytest.mark.skipif(
isinstance(get_default_mode().linker, NumbaLinker),
reason="Running with numba linker, c backend (should be) covered in another CI",
)
def test_c_nan(self): def test_c_nan(self):
for dtype in ["floatX", "complex64", "complex128"]: for dtype in ["floatX", "complex64", "complex128"]:
self.with_mode(Mode(linker="c"), ps.add, dtype=dtype, test_nan=True) self.with_mode(Mode(linker="c"), ps.add, dtype=dtype, test_nan=True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论