set no_gpu True

apache · Nov 20, 2020 · df8a3e8 · df8a3e8
1 parent 7ae26ed
commit df8a3e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
@@ -3628,7 +3628,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)
@@ -3674,7 +3674,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)