From 66839b405c52e30f5ef6e5c484662ab783faa895 Mon Sep 17 00:00:00 2001 From: xp56 Date: Thu, 19 Nov 2020 20:10:27 +0800 Subject: [PATCH] set no_gpu True --- tests/python/frontend/tensorflow/test_forward.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py index 4fb721accf016..ec15b019fc0dd 100644 --- a/tests/python/frontend/tensorflow/test_forward.py +++ b/tests/python/frontend/tensorflow/test_forward.py @@ -3665,7 +3665,7 @@ def _test_forward_segment(name, data_shape, size): tf_segment_ids = tf.constant(segment_ids) segment_op = getattr(tf.math, name) result = segment_op(in_data, tf_segment_ids, name=name) - compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, False) + compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, no_gpu=True) _test_forward_segment("segment_max", [100], 20) _test_forward_segment("segment_max", [10, 3, 4], 5) @@ -3711,7 +3711,7 @@ def _test_forward_unsorted_segment(name, data_shape, size): tf_segment_ids = tf.constant(segment_ids) segment_op = getattr(tf.math, name) result = segment_op(in_data, tf_segment_ids, num_segments, name=name) - compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, False) + compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, no_gpu=True) _test_forward_unsorted_segment("unsorted_segment_max", [100], 20) _test_forward_unsorted_segment("unsorted_segment_max", [10, 3, 4], 5)