From 21380d856aedaf4f298b4c73b3e7220830b137de Mon Sep 17 00:00:00 2001 From: lcskrishna Date: Fri, 31 Aug 2018 13:33:54 -0700 Subject: [PATCH] skipping all the datatypes with skipIfRocm --- test/test_cuda.py | 66 ++++++++++++++++------------------------------- 1 file changed, 22 insertions(+), 44 deletions(-) diff --git a/test/test_cuda.py b/test/test_cuda.py index 03a2ff5af641fe..164b3859fea4d4 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -347,26 +347,19 @@ def tmp(t): ('kthvalue', small_3d_unique, lambda t: [3, -1], 'neg_dim'), ('lerp', small_3d, lambda t: [small_3d(t), 0.3], '', types, False, "skipIfRocm:HalfTensor"), ('max', small_3d_unique, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), - ('max', small_3d_unique, lambda t: [1], 'dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('max', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('max', small_3d_unique, lambda t: [1], 'dim', types, False, skipIfRocm), + ('max', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, skipIfRocm), ('max', medium_2d, lambda t: [medium_2d(t)], 'elementwise'), ('min', small_3d_unique, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), - ('min', small_3d_unique, lambda t: [1], 'dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('min', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('min', small_3d_unique, lambda t: [1], 'dim', types, False, skipIfRocm), + ('min', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, skipIfRocm), ('min', medium_2d, lambda t: [medium_2d(t)], 'elementwise'), ('mean', small_3d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), ('mean', small_3d, lambda t: [-1], 'neg_dim', types, False, "skipIfRocm:DoubleTensor,FloatTensor,HalfTensor"), ('mean', small_3d, lambda t: [1], 'dim', types, False, "skipIfRocm:DoubleTensor,FloatTensor,HalfTensor"), - ('mode', small_3d, lambda t: [], '', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('mode', small_3d, lambda t: [1], 'dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('mode', small_3d, lambda t: [-1], 'neg_dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('mode', small_3d, lambda t: [], '', types, False, skipIfRocm), + ('mode', small_3d, lambda t: [1], 'dim', types, False, skipIfRocm), + ('mode', small_3d, lambda t: [-1], 'neg_dim', types, False, skipIfRocm), ('mvlgamma', lambda t: tensor_clamp(small_2d(t), 0.1, 10), lambda t: [1], '2d_p=1', float_types_no_half, False, "skipIfRocm:DoubleTensor,FloatTensor"), ('mvlgamma', lambda t: tensor_clamp(small_2d(t), 0.6, 10), lambda t: [2], '2d_p=2', float_types_no_half, @@ -386,8 +379,7 @@ def tmp(t): ('numel', small_3d, lambda t: [],), ('narrow', small_3d, lambda t: [1, 3, 2],), ('narrow', small_3d, lambda t: [-1, 3, 2], 'neg_dim'), - ('nonzero', small_3d, lambda t: [], '', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('nonzero', small_3d, lambda t: [], '', types, False, skipIfRocm), ('norm', small_3d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), ('norm', small_3d, lambda t: [3], '3_norm', types, False, "skipIfRocm:HalfTensor"), ('norm', small_3d, lambda t: [3, 0], '3_norm_dim', types, False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"), @@ -395,20 +387,15 @@ def tmp(t): False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"), ('ones', small_3d, lambda t: [1, 2, 3, 4, 5],), ('permute', new_t(1, 2, 3, 4), lambda t: [2, 1, 3, 0],), - ('put_', new_t(2, 5, 3), lambda t: [long_type(t)([[0], [-2]]), t([[3], [4]])], '', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('put_', new_t(2, 5, 3), lambda t: [long_type(t)([[0], [-2]]), t([[3], [4]])], '', types, False, skipIfRocm), ('put_', new_t(2, 3), lambda t: [long_type(t)([]), t([])], 'empty'), ('put_', new_t(2, 2), lambda t: [long_type(t)([[1], [-3]]), t([[1], [2]]), True], 'accumulate'), ('prod', small_2d_oneish, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), - ('prod', small_3d, lambda t: [1], 'dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('prod', small_3d, lambda t: [-1], 'neg_dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('prod', small_3d, lambda t: [1], 'dim', types, False, skipIfRocm), + ('prod', small_3d, lambda t: [-1], 'neg_dim', types, False, skipIfRocm), ('sum', small_2d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), - ('sum', small_3d, lambda t: [1], 'dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('sum', small_3d, lambda t: [-1], 'neg_dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('sum', small_3d, lambda t: [1], 'dim', types, False, skipIfRocm), + ('sum', small_3d, lambda t: [-1], 'neg_dim', types, False, skipIfRocm), ('renorm', small_3d, lambda t: [2, 1, 1], '2_norm', types, False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"), ('renorm', small_3d, lambda t: [2, -1, 1], '2_norm_neg_dim', types, False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"), @@ -418,16 +405,11 @@ def tmp(t): ('size', new_t(1, 2, 3, 4), lambda t: [],), ('size', new_t(1, 2, 3, 4), lambda t: [1], 'dim'), ('size', new_t(1, 2, 3, 4), lambda t: [-2], 'neg_dim'), - ('sort', small_3d_unique, lambda t: [], '', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('sort', small_3d_unique, lambda t: [1], 'dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('sort', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('sort', small_3d_unique, lambda t: [1, True], 'dim_descending', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('sort', small_3d_unique, lambda t: [-1, True], 'neg_dim_descending', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('sort', small_3d_unique, lambda t: [], '', types, False, skipIfRocm), + ('sort', small_3d_unique, lambda t: [1], 'dim', types, False, skipIfRocm), + ('sort', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, skipIfRocm), + ('sort', small_3d_unique, lambda t: [1, True], 'dim_descending', types, False, skipIfRocm), + ('sort', small_3d_unique, lambda t: [-1, True], 'neg_dim_descending', types, False, skipIfRocm), ('split', small_3d, lambda t: [2],), ('split', small_3d, lambda t: [2, 1], 'dim'), ('split', small_3d, lambda t: [2, -3], 'neg_dim'), @@ -435,17 +417,13 @@ def tmp(t): ('squeeze', new_t(1, 2, 1, 4), lambda t: [2], 'dim'), ('squeeze', new_t(1, 2, 1, 4), lambda t: [-2], 'neg_dim'), ('t', new_t(1, 2), lambda t: [],), - ('take', new_t(3, 4), lambda t: [long_type(t)([[0], [-2]])], '', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), + ('take', new_t(3, 4), lambda t: [long_type(t)([[0], [-2]])], '', types, False, skipIfRocm), ('transpose', new_t(1, 2, 3, 4), lambda t: [1, 2],), ('transpose', new_t(1, 2, 3, 4), lambda t: [-1, -2], 'neg_dim'), ('to_list', small_3d, lambda t: [],), - ('topk', small_3d_unique, lambda t: [2, 1, False, True], 'dim_sort', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('topk', small_3d_unique, lambda t: [2, -1, False, True], 'neg_dim_sort', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"), - ('topk', small_3d_unique, lambda t: [2, 1, True, True], 'dim_desc_sort', types, False, - "skipIfRocm:ByteTensor,CharTensor,DoubleTensor,HalfTensor,LongTensor,FloatTensor,IntTensor,ShortTensor"), + ('topk', small_3d_unique, lambda t: [2, 1, False, True], 'dim_sort', types, False, skipIfRocm), + ('topk', small_3d_unique, lambda t: [2, -1, False, True], 'neg_dim_sort', types, False, skipIfRocm), + ('topk', small_3d_unique, lambda t: [2, 1, True, True], 'dim_desc_sort', types, False, skipIfRocm), ('trace', medium_2d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"), ('tril', medium_2d, lambda t: [],), ('tril', medium_2d_expanded, lambda t: [], 'zero_stride', types, True),