Add test_forward_index_put to __main__ (apache#7542)

Lokiiiiii · Mar 2, 2021 · 35b9a43 · 35b9a43
1 parent 497d5e5
commit 35b9a43
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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