File tree 2 files changed +12
-6
lines changed
2 files changed +12
-6
lines changed Original file line number Diff line number Diff line change 31
31
torch .int32 ,
32
32
torch .int64 ,
33
33
}
34
- try :
35
- # torch >=2.3
36
- _int_dtypes |= {torch .uint16 , torch .uint32 , torch .uint64 }
37
- except AttributeError :
38
- pass
39
-
40
34
41
35
_array_api_dtypes = {
42
36
torch .bool ,
Original file line number Diff line number Diff line change @@ -8,11 +8,23 @@ array_api_tests/test_array_object.py::test_getitem
8
8
array_api_tests/test_array_object.py::test_setitem
9
9
# Masking doesn't suport 0 dimensions in the mask
10
10
array_api_tests/test_array_object.py::test_getitem_masking
11
+ # torch doesn't have uint dtypes other than uint8
12
+ array_api_tests/test_array_object.py::test_scalar_casting[__int__(uint16)]
13
+ array_api_tests/test_array_object.py::test_scalar_casting[__int__(uint32)]
14
+ array_api_tests/test_array_object.py::test_scalar_casting[__int__(uint64)]
15
+ array_api_tests/test_array_object.py::test_scalar_casting[__index__(uint16)]
16
+ array_api_tests/test_array_object.py::test_scalar_casting[__index__(uint32)]
17
+ array_api_tests/test_array_object.py::test_scalar_casting[__index__(uint64)]
11
18
12
19
# Overflow error from large inputs
13
20
array_api_tests/test_creation_functions.py::test_arange
14
21
# pytorch linspace bug (should be fixed in torch 2.0)
15
22
23
+ # torch doesn't have higher uint dtypes
24
+ array_api_tests/test_data_type_functions.py::test_iinfo[uint16]
25
+ array_api_tests/test_data_type_functions.py::test_iinfo[uint32]
26
+ array_api_tests/test_data_type_functions.py::test_iinfo[uint64]
27
+
16
28
# We cannot wrap the tensor object
17
29
array_api_tests/test_has_names.py::test_has_names[array_method-__array_namespace__]
18
30
array_api_tests/test_has_names.py::test_has_names[array_method-to_device]
You can’t perform that action at this time.
0 commit comments