From 113094c63c0e5d099c66e963ce434a490de3d3b7 Mon Sep 17 00:00:00 2001
From: Trevor Morris <trevmorr@amazon.com>
Date: Mon, 15 Jun 2020 19:31:55 +0000
Subject: [PATCH] Increase stack limit for failing tflite tests. Skip TF tests
 which require TF 1.x

---
 docker/bash.sh                                   |  1 +
 tests/python/frontend/tensorflow/test_forward.py | 15 +++++++++++++++
 tests/python/frontend/tflite/test_forward.py     |  2 --
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/docker/bash.sh b/docker/bash.sh
index 008aa6a9a30c..d46ee18b34a9 100755
--- a/docker/bash.sh
+++ b/docker/bash.sh
@@ -83,6 +83,7 @@ ${DOCKER_BINARY} run --rm --pid=host\
     -v ${WORKSPACE}:/workspace \
     -v ${SCRIPT_DIR}:/docker \
     -w /workspace \
+    --ulimit stack=16777216:16777216 \
     -e "CI_BUILD_HOME=/workspace" \
     -e "CI_BUILD_USER=$(id -u -n)" \
     -e "CI_BUILD_UID=$(id -u)" \
diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
index e78f9d0cb6e7..252c4cd8904a 100644
--- a/tests/python/frontend/tensorflow/test_forward.py
+++ b/tests/python/frontend/tensorflow/test_forward.py
@@ -873,6 +873,9 @@ def test_forward_squeeze():
 # TensorArray
 # -----------
 def test_tensor_array_write_read():
+    if package_version.parse(tf.VERSION) >= package_version.parse('1.15.0'):
+        pytest.skip("Needs fixing for tflite >= 1.15.0")
+
     def run(dtype_str, infer_shape, element_shape):
         with tf.Graph().as_default():
             dtype = tf_dtypes[dtype_str]
@@ -895,6 +898,9 @@ def run(dtype_str, infer_shape, element_shape):
 
 
 def test_tensor_array_scatter():
+    if package_version.parse(tf.VERSION) >= package_version.parse('1.15.0'):
+        pytest.skip("Needs fixing for tflite >= 1.15.0")
+
     def run(dtype_str, infer_shape):
         with tf.Graph().as_default():
             dtype =  tf_dtypes[dtype_str]
@@ -921,6 +927,9 @@ def run(dtype_str, infer_shape):
 
 
 def test_tensor_array_gather():
+    if package_version.parse(tf.VERSION) >= package_version.parse('1.15.0'):
+        pytest.skip("Needs fixing for tflite >= 1.15.0")
+
     def run(dtype_str, infer_shape):
         with tf.Graph().as_default():
             dtype =  tf_dtypes[dtype_str]
@@ -937,6 +946,9 @@ def run(dtype_str, infer_shape):
 
 
 def test_tensor_array_split():
+    if package_version.parse(tf.VERSION) >= package_version.parse('1.15.0'):
+        pytest.skip("Needs fixing for tflite >= 1.15.0")
+
     def run(dtype_str, infer_shape):
         with tf.Graph().as_default():
             dtype =  tf_dtypes[dtype_str]
@@ -959,6 +971,9 @@ def run(dtype_str, infer_shape):
 
 
 def test_tensor_array_concat():
+    if package_version.parse(tf.VERSION) >= package_version.parse('1.15.0'):
+        pytest.skip("Needs fixing for tflite >= 1.15.0")
+
     def run(dtype_str, infer_shape):
         with tf.Graph().as_default():
             dtype = tf_dtypes[dtype_str]
diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py
index 004ca40f007f..29515403c6b6 100644
--- a/tests/python/frontend/tflite/test_forward.py
+++ b/tests/python/frontend/tflite/test_forward.py
@@ -2286,7 +2286,6 @@ def test_forward_qnn_mobilenet_v1_net():
     tvm_sorted_labels = tvm_predictions.argsort()[-3:][::-1]
     tvm.testing.assert_allclose(tvm_sorted_labels, tflite_sorted_labels)
 
-@pytest.mark.skip("neo-ai/tvm: disabled due to segfault in CI")
 def test_forward_qnn_mobilenet_v2_net():
     """Test the Quantized TFLite Mobilenet V2 model."""
     # MobilenetV2
@@ -2439,7 +2438,6 @@ def test_forward_coco_ssd_mobilenet_v1():
 # MediaPipe
 # -------------
 
-@pytest.mark.skip("neo-ai/tvm: disabled due to error in CI")
 def test_forward_mediapipe_hand_landmark():
     """Test MediaPipe 2D hand landmark TF Lite model."""
     # MediaPipe 2D hand landmark TF