Skip to content

Commit 8fa5824

Browse files
committed
Remove fast_run from mode in test_shape_i_const
`fast_run` includes the fusion optimization, but that should not be called with a `fast_compile` mode because that prevents the creation of `c_thunks`, even when a `C` compiler is available. This forces the use of Python perform method which is limited to 32 operands, when the Fusion Optimizer assumed that the C limit (1024 operands) was in place. In this test, one of the fused Composite surpasses that limit.
1 parent 22c5db3 commit 8fa5824

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tensor/test_subtensor.py

+1-1
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 = []

0 commit comments

Comments
 (0)