Skip to content

Commit dbe0e09

Browse files
aseyboldtricardoV94
authored andcommitted
fix(numba): Remove useless blockwise in tests
1 parent d92c367 commit dbe0e09

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/link/numba/test_basic.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,12 @@ def perform(self, node, inputs, outputs):
118118
my_multi_out.ufunc = MyMultiOut.impl
119119
my_multi_out.ufunc.nin = 2
120120
my_multi_out.ufunc.nout = 2
121-
opts = RewriteDatabaseQuery(include=[None], exclude=["cxx_only", "BlasOpt"])
122-
numba_mode = Mode(NumbaLinker(), opts.including("numba"))
121+
opts = RewriteDatabaseQuery(
122+
include=[None], exclude=["cxx_only", "BlasOpt", "local_careduce_fusion"]
123+
)
124+
numba_mode = Mode(
125+
NumbaLinker(), opts.including("numba", "local_useless_unbatched_blockwise")
126+
)
123127
py_mode = Mode("py", opts)
124128

125129
rng = np.random.default_rng(42849)

0 commit comments

Comments
 (0)