Skip to content

Commit 46a2227

Browse files
committed
Hard-code the default torch integral type to int64
See the discussion at #166 (comment)
1 parent 176a66a commit 46a2227

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

array_api_compat/torch/_info.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def default_dtypes(self, *, device=None):
149149
"""
150150
default_floating = torch.get_default_dtype()
151151
default_complex = torch.complex64 if default_floating == torch.float32 else torch.complex128
152-
default_integral = torch.asarray(0, device=device).dtype
152+
default_integral = torch.int64
153153
return {
154154
"real floating": default_floating,
155155
"complex floating": default_complex,

0 commit comments

Comments
 (0)