-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Open
Labels
high prioritymodule: python array apiIssues related to the Python Array APIIssues related to the Python Array APItrackerA tracking issueA tracking issuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
Progress tracker for all compatibility issues with Python Array API specification:
-
__array_namespace__(add at/towards the end, it's the attribute that declares compliance) - Support for
uint16,uint32, anduint64#58734 - type promotion with 0d-tensors diverges from array API specification #58736
- Python Array API: support constants #58739
- support for scalars in
logical_(and|not|or|xor)#58740 -
torch.size()diverges from array API specification #58741 - Python Array API New Operators Tracker #58742
- Implement improved DLPack support #55090
-
torch.(min|max)(..., dim=...)diverges from array API specification #58745 -
torch.nonzerodiverges from the specification (it returns a tensor, not a tuple, by default) - see [numpy] torch.nonzero is similar to np.argwhere not np.nonzero #64502 - Support negative step sizes for slicing #59786
- Remove support for multiple ellipses in slicing #59787
-
Bitwise shifts should retain the same dtype as the first input #59867 - Python scalars should be promoted to the same
dtypeas the respective tensor #59868 -
torch.asarraydoes not detect dtype of Python scalars #70591 -
fullshould take an integer size #70906 -
finfo(...).tinyshould be aliased tofinfo(...).smallest_normal#70909 -
eyeshould support other diagonals than the main one #70910 - support setting
stepinarangewithout settingend#70914 -
arangeshould return empty array if bounds are inconsistent with step sign #70915 - uint8 scalar tensors cannot be used for integer indexing #70916
- make
torch.(ceil|floor|round|trunc)no-ops for integer inputs #70918 -
linspaceshould support anendpointparameter #70919 -
uniqueshould be split into four partial functions #70920 -
sortshould only return the sorted input #70921 -
argsortis missing thestableparameter #70922 -
torch.whereto support Python scalars and type promotion #9190 - change supported arguments for parameter
diminsqueeze#70924 - allow
dim=Noneinconcat#70925
cc @ezyang @gchanan @zou3519 @bdhirsh @jbschlosser @anjali411
kmaehashi, asi1024, xpmemeda, aregm, EdAyers and 8 moremruberry, anjali411, xpmemeda, EdAyers, alisterburt and 5 more
Metadata
Metadata
Assignees
Labels
high prioritymodule: python array apiIssues related to the Python Array APIIssues related to the Python Array APItrackerA tracking issueA tracking issuetriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module