From 7b305ed1b0bcb940ab3d2692fa8be836260edfd2 Mon Sep 17 00:00:00 2001 From: Alexander Pivovarov Date: Fri, 26 Feb 2021 18:28:02 -0800 Subject: [PATCH] Add test_forward_index_put to __main__ (#7542) --- python/tvm/auto_scheduler/feature.py | 2 +- tests/python/frontend/pytorch/test_forward.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/python/tvm/auto_scheduler/feature.py b/python/tvm/auto_scheduler/feature.py index bd6526187581..ec7cf6334f98 100644 --- a/python/tvm/auto_scheduler/feature.py +++ b/python/tvm/auto_scheduler/feature.py @@ -120,7 +120,7 @@ def unpack_feature(byte_arr: bytearray) -> Tuple[np.ndarray, np.ndarray, np.ndar tmp_vec_len = (size - 1) // n_stmts assert ( tmp_vec_len == vec_len - ), "The lenght of feature vector is wrong. " "Expected %d but got %d." % ( + ), "The length of feature vector is wrong. Expected %d but got %d." % ( vec_len, tmp_vec_len, ) diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py index 90604751d4f1..6acd8b299920 100644 --- a/tests/python/frontend/pytorch/test_forward.py +++ b/tests/python/frontend/pytorch/test_forward.py @@ -1957,7 +1957,7 @@ def _impl(inputs, input_types): @tvm.testing.uses_gpu -def test_segmentaton_models(): +def test_segmentation_models(): class SegmentationModelWrapper(Module): def __init__(self, model): super().__init__() @@ -3811,6 +3811,7 @@ def test_fn(is_sorted, return_inverse, return_counts): test_forward_unbind() test_forward_nonzero() test_forward_scatter() + test_forward_index_put() test_numel() test_bincount() test_cumsum() @@ -3836,7 +3837,7 @@ def test_fn(is_sorted, return_inverse, return_counts): test_custom_conversion_map() - test_segmentaton_models() + test_segmentation_models() test_3d_models() # Quantization test