Skip to content

Commit 3ca46e8

Browse files
committed
fix concatenate tests failing due to not being bit exact
Since requantize is now offloaded, concatenate tests were failing due a reference not being used. Change-Id: I44b26b5daecfefb776ca19e6646f3690f5570f52
1 parent 2cdcc85 commit 3ca46e8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/python/contrib/test_ethosu/test_codegen.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -950,7 +950,10 @@ def concat_func(*inputs):
950950
op = tf.concat(list(inputs), axis)
951951
return op
952952

953-
_compare_tvm_with_tflite(concat_func, shapes, accel_type)
953+
# TODO(lhutton1) For now output is not bit exact with TFLite.
954+
# This is because TFLite reference kernels are not being used.
955+
# For this, TFLite will need upgrading to 2.6.
956+
_compare_tvm_with_tflite(concat_func, shapes, accel_type, output_tolerance=1)
954957

955958

956959
@pytest.mark.parametrize("accel_type", ACCEL_TYPES)

0 commit comments

Comments
 (0)