Skip to content

Commit ba34557

Browse files
abhudevkulinseth
authored andcommitted
Cast dot inputs to int32 when needed (#121)
* Cast dot inputs to int32 when needed * Add dot uint8 to allowlist * Add support for int8 input * Fix indentation
1 parent 1d5156c commit ba34557

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/test_mps.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7821,6 +7821,10 @@ class TestConsistency(TestCase):
78217821
'trapz': ['torch.float16', 'torch.float32', 'torch.int16', 'torch.int32', 'torch.int64', 'torch.uint8'],
78227822
'nn.functional.local_response_norm': ['torch.int64'],
78237823
'index_select': ['torch.bool'],
7824+
'repeat': ['torch.bool'],
7825+
'rot90': ['torch.bool'],
7826+
'tile': ['torch.bool'],
7827+
'split': ['torch.float32'],
78247828
}
78257829

78267830
# Used for accept mode only

0 commit comments

Comments
 (0)