Skip to content

Commit a6500fa

Browse files
committed
Set "max dimensions" to None
`usm_ndarray` can support potentially arbitrary dimensions
1 parent 29eeac7 commit a6500fa

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

dpctl/tensor/_array_api.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def __init__(self):
7575
self._capabilities = {
7676
"boolean indexing": True,
7777
"data-dependent shapes": True,
78-
"max dimensions": 64,
78+
"max dimensions": None,
7979
}
8080
self._all_dtypes = {
8181
"bool": dpt.bool,
@@ -108,7 +108,7 @@ def capabilities(self):
108108
Value: ``True``
109109
``max dimensions``:
110110
integer indication the maximum array dimension supported by ``dpctl``.
111-
Value: ``64``
111+
Value: ``None``
112112
113113
Returns:
114114
dict:

dpctl/tests/test_tensor_array_api_inspection.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def test_array_api_inspection_capabilities():
7474
capabilities = dpt.__array_namespace_info__().capabilities()
7575
assert capabilities["boolean indexing"]
7676
assert capabilities["data-dependent shapes"]
77-
assert capabilities["max dimensions"] == 64
77+
assert capabilities["max dimensions"] is None
7878

7979

8080
def test_array_api_inspection_default_dtypes():

0 commit comments

Comments
 (0)