Skip to content

Commit 321da01

Browse files
committed
Do not manually include fast_run in test_shape_i_*
It doesn't make sense to include `fast_run` if `fast_compile` mode is being used. Some rewrites such as the FusionOptimizer are not compatible with `fast_compile` mode which prevents the creation of C thunks. The FusionOptimizer has no way of knowing this is the case, and assumes it is safe to return Composites with more than 32 operands, even though that's not the case with the Python perform method.
1 parent c73f5cc commit 321da01

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/tensor/test_subtensor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -986,7 +986,7 @@ def test_adv_sub1_idx_broadcast(self):
986986
def test_shape_i_const(self):
987987
# Each axis is treated independently by shape_i/shape operators
988988

989-
mode_opt = self.mode.including("fast_run")
989+
mode_opt = self.mode
990990
data = self.shared(np.array(np.arange(5), dtype=self.dtype))
991991
for start in [None] + [-8, -5, -1, 0, 1, 5, 8]:
992992
outs = []
@@ -1004,7 +1004,7 @@ def test_shape_i_const(self):
10041004
def test_shape_i_scalar(self):
10051005
# Each axis is treated independently by shape_i/shape operators
10061006

1007-
mode_opt = self.mode.including("fast_run")
1007+
mode_opt = self.mode
10081008

10091009
v_data = np.array(np.arange(5), dtype=self.dtype)
10101010
t_data = self.shared(v_data)

0 commit comments

Comments
 (0)