Skip to content

Commit 734d988

Browse files
iotamudeltaiotamudelta
authored
Merge pull request #166 from lcskrishna/cl/code-review
test_cuda - skipping all the types with skipIfRocm
2 parents 622f720 + 21380d8 commit 734d988

File tree

1 file changed

+22
-44
lines changed

1 file changed

+22
-44
lines changed

‎test/test_cuda.py

Lines changed: 22 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -347,26 +347,19 @@ def tmp(t):
347347
('kthvalue', small_3d_unique, lambda t: [3, -1], 'neg_dim'),
348348
('lerp', small_3d, lambda t: [small_3d(t), 0.3], '', types, False, "skipIfRocm:HalfTensor"),
349349
('max', small_3d_unique, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
350-
('max', small_3d_unique, lambda t: [1], 'dim', types, False,
351-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
352-
('max', small_3d_unique, lambda t: [-1], 'neg_dim', types, False,
353-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
350+
('max', small_3d_unique, lambda t: [1], 'dim', types, False, skipIfRocm),
351+
('max', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, skipIfRocm),
354352
('max', medium_2d, lambda t: [medium_2d(t)], 'elementwise'),
355353
('min', small_3d_unique, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
356-
('min', small_3d_unique, lambda t: [1], 'dim', types, False,
357-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
358-
('min', small_3d_unique, lambda t: [-1], 'neg_dim', types, False,
359-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
354+
('min', small_3d_unique, lambda t: [1], 'dim', types, False, skipIfRocm),
355+
('min', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, skipIfRocm),
360356
('min', medium_2d, lambda t: [medium_2d(t)], 'elementwise'),
361357
('mean', small_3d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
362358
('mean', small_3d, lambda t: [-1], 'neg_dim', types, False, "skipIfRocm:DoubleTensor,FloatTensor,HalfTensor"),
363359
('mean', small_3d, lambda t: [1], 'dim', types, False, "skipIfRocm:DoubleTensor,FloatTensor,HalfTensor"),
364-
('mode', small_3d, lambda t: [], '', types, False,
365-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
366-
('mode', small_3d, lambda t: [1], 'dim', types, False,
367-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
368-
('mode', small_3d, lambda t: [-1], 'neg_dim', types, False,
369-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
360+
('mode', small_3d, lambda t: [], '', types, False, skipIfRocm),
361+
('mode', small_3d, lambda t: [1], 'dim', types, False, skipIfRocm),
362+
('mode', small_3d, lambda t: [-1], 'neg_dim', types, False, skipIfRocm),
370363
('mvlgamma', lambda t: tensor_clamp(small_2d(t), 0.1, 10), lambda t: [1], '2d_p=1', float_types_no_half,
371364
False, "skipIfRocm:DoubleTensor,FloatTensor"),
372365
('mvlgamma', lambda t: tensor_clamp(small_2d(t), 0.6, 10), lambda t: [2], '2d_p=2', float_types_no_half,
@@ -386,29 +379,23 @@ def tmp(t):
386379
('numel', small_3d, lambda t: [],),
387380
('narrow', small_3d, lambda t: [1, 3, 2],),
388381
('narrow', small_3d, lambda t: [-1, 3, 2], 'neg_dim'),
389-
('nonzero', small_3d, lambda t: [], '', types, False,
390-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
382+
('nonzero', small_3d, lambda t: [], '', types, False, skipIfRocm),
391383
('norm', small_3d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
392384
('norm', small_3d, lambda t: [3], '3_norm', types, False, "skipIfRocm:HalfTensor"),
393385
('norm', small_3d, lambda t: [3, 0], '3_norm_dim', types, False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"),
394386
('norm', small_3d, lambda t: [3, -2], '3_norm_neg_dim', types,
395387
False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"),
396388
('ones', small_3d, lambda t: [1, 2, 3, 4, 5],),
397389
('permute', new_t(1, 2, 3, 4), lambda t: [2, 1, 3, 0],),
398-
('put_', new_t(2, 5, 3), lambda t: [long_type(t)([[0], [-2]]), t([[3], [4]])], '', types, False,
399-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
390+
('put_', new_t(2, 5, 3), lambda t: [long_type(t)([[0], [-2]]), t([[3], [4]])], '', types, False, skipIfRocm),
400391
('put_', new_t(2, 3), lambda t: [long_type(t)([]), t([])], 'empty'),
401392
('put_', new_t(2, 2), lambda t: [long_type(t)([[1], [-3]]), t([[1], [2]]), True], 'accumulate'),
402393
('prod', small_2d_oneish, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
403-
('prod', small_3d, lambda t: [1], 'dim', types, False,
404-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
405-
('prod', small_3d, lambda t: [-1], 'neg_dim', types, False,
406-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
394+
('prod', small_3d, lambda t: [1], 'dim', types, False, skipIfRocm),
395+
('prod', small_3d, lambda t: [-1], 'neg_dim', types, False, skipIfRocm),
407396
('sum', small_2d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
408-
('sum', small_3d, lambda t: [1], 'dim', types, False,
409-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
410-
('sum', small_3d, lambda t: [-1], 'neg_dim', types, False,
411-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
397+
('sum', small_3d, lambda t: [1], 'dim', types, False, skipIfRocm),
398+
('sum', small_3d, lambda t: [-1], 'neg_dim', types, False, skipIfRocm),
412399
('renorm', small_3d, lambda t: [2, 1, 1], '2_norm', types, False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"),
413400
('renorm', small_3d, lambda t: [2, -1, 1], '2_norm_neg_dim', types,
414401
False, "skipIfRocm:HalfTensor,DoubleTensor,FloatTensor"),
@@ -418,34 +405,25 @@ def tmp(t):
418405
('size', new_t(1, 2, 3, 4), lambda t: [],),
419406
('size', new_t(1, 2, 3, 4), lambda t: [1], 'dim'),
420407
('size', new_t(1, 2, 3, 4), lambda t: [-2], 'neg_dim'),
421-
('sort', small_3d_unique, lambda t: [], '', types, False,
422-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
423-
('sort', small_3d_unique, lambda t: [1], 'dim', types, False,
424-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
425-
('sort', small_3d_unique, lambda t: [-1], 'neg_dim', types, False,
426-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
427-
('sort', small_3d_unique, lambda t: [1, True], 'dim_descending', types, False,
428-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
429-
('sort', small_3d_unique, lambda t: [-1, True], 'neg_dim_descending', types, False,
430-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
408+
('sort', small_3d_unique, lambda t: [], '', types, False, skipIfRocm),
409+
('sort', small_3d_unique, lambda t: [1], 'dim', types, False, skipIfRocm),
410+
('sort', small_3d_unique, lambda t: [-1], 'neg_dim', types, False, skipIfRocm),
411+
('sort', small_3d_unique, lambda t: [1, True], 'dim_descending', types, False, skipIfRocm),
412+
('sort', small_3d_unique, lambda t: [-1, True], 'neg_dim_descending', types, False, skipIfRocm),
431413
('split', small_3d, lambda t: [2],),
432414
('split', small_3d, lambda t: [2, 1], 'dim'),
433415
('split', small_3d, lambda t: [2, -3], 'neg_dim'),
434416
('squeeze', new_t(1, 2, 1, 4), lambda t: [],),
435417
('squeeze', new_t(1, 2, 1, 4), lambda t: [2], 'dim'),
436418
('squeeze', new_t(1, 2, 1, 4), lambda t: [-2], 'neg_dim'),
437419
('t', new_t(1, 2), lambda t: [],),
438-
('take', new_t(3, 4), lambda t: [long_type(t)([[0], [-2]])], '', types, False,
439-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
420+
('take', new_t(3, 4), lambda t: [long_type(t)([[0], [-2]])], '', types, False, skipIfRocm),
440421
('transpose', new_t(1, 2, 3, 4), lambda t: [1, 2],),
441422
('transpose', new_t(1, 2, 3, 4), lambda t: [-1, -2], 'neg_dim'),
442423
('to_list', small_3d, lambda t: [],),
443-
('topk', small_3d_unique, lambda t: [2, 1, False, True], 'dim_sort', types, False,
444-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
445-
('topk', small_3d_unique, lambda t: [2, -1, False, True], 'neg_dim_sort', types, False,
446-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,FloatTensor,HalfTensor,IntTensor,LongTensor,ShortTensor"),
447-
('topk', small_3d_unique, lambda t: [2, 1, True, True], 'dim_desc_sort', types, False,
448-
"skipIfRocm:ByteTensor,CharTensor,DoubleTensor,HalfTensor,LongTensor,FloatTensor,IntTensor,ShortTensor"),
424+
('topk', small_3d_unique, lambda t: [2, 1, False, True], 'dim_sort', types, False, skipIfRocm),
425+
('topk', small_3d_unique, lambda t: [2, -1, False, True], 'neg_dim_sort', types, False, skipIfRocm),
426+
('topk', small_3d_unique, lambda t: [2, 1, True, True], 'dim_desc_sort', types, False, skipIfRocm),
449427
('trace', medium_2d, lambda t: [], '', types, False, "skipIfRocm:HalfTensor"),
450428
('tril', medium_2d, lambda t: [],),
451429
('tril', medium_2d_expanded, lambda t: [], 'zero_stride', types, True),

0 commit comments

Comments
 (0)