-
Notifications
You must be signed in to change notification settings - Fork 27.7k
Python Array API Compatibility Tracker #58743
Copy link
Copy link
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
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
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)uint16,uint32, anduint64#58734logical_(and|not|or|xor)#58740torch.size()diverges from array API specification #58741torch.(min|max)(..., dim=...)diverges from array API specification #58745torch.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 #64502Bitwise shifts should retain the same dtype as the first input #59867dtypeas the respective tensor #59868torch.asarraydoes not detect dtype of Python scalars #70591fullshould take an integer size #70906finfo(...).tinyshould be aliased tofinfo(...).smallest_normal#70909eyeshould support other diagonals than the main one #70910stepinarangewithout settingend#70914arangeshould return empty array if bounds are inconsistent with step sign #70915torch.(ceil|floor|round|trunc)no-ops for integer inputs #70918linspaceshould support anendpointparameter #70919uniqueshould be split into four partial functions #70920sortshould only return the sorted input #70921argsortis missing thestableparameter #70922torch.whereto support Python scalars and type promotion #9190diminsqueeze#70924dim=Noneinconcat#70925cc @ezyang @gchanan @zou3519 @bdhirsh @jbschlosser @anjali411