tvm
tvm copied to clipboard
[BUG] List index out of range compiling `test_tflite_large_irregular` on `arm_cpu` target
trafficstars
Expected behaviour:
The test tests/python/relay/test_op_qnn_conv2d.py:test_tflite_large_irregular runs successfully when the target is arm_cpu.
Actual behaviour:
The test fails to run and gives the following error:
python/tvm/autotvm/task/space.py:736: in define_split
return self._add_new_transform(SplitSpace, name, axes, policy, **kwargs)
python/tvm/autotvm/task/space.py:1132: in _add_new_transform
self._entity_map[name] = space[0]
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _
self = Split(policy=factors, product=1001, num_outputs=2) len=0, index = 0
def __getitem__(self, index):
"""Get an entity of the space by index
Parameters
----------
index: int
Returns
-------
transform entity
"""
> return self.entities[index]
E IndexError: list index out of range
python/tvm/autotvm/task/space.py:93: IndexError
Environment:
Tested with TVM at https://github.com/apache/tvm/commit/6a3fadc0654ecf9557ffe08d24677684c96e80b0. The issue was found as a result of the changes in https://github.com/apache/tvm/pull/16513, however it can be reproduced without as described below.
How to reproduce:
Run pytest tests/python/relay/test_op_qnn_conv2d.py -k test_tflite_large_irregular with an arm_cpu target. Note: Reminder to remove any skip condition that exists in the test currently.