From a6500fa18a80b1198578f5727fc2c31bd27c141b Mon Sep 17 00:00:00 2001 From: Nikita Grigorian Date: Wed, 30 Apr 2025 20:52:38 -0700 Subject: [PATCH] Set `"max dimensions"` to `None` `usm_ndarray` can support potentially arbitrary dimensions --- dpctl/tensor/_array_api.py | 4 ++-- dpctl/tests/test_tensor_array_api_inspection.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/dpctl/tensor/_array_api.py b/dpctl/tensor/_array_api.py index 8aade1ca47..ad798d4327 100644 --- a/dpctl/tensor/_array_api.py +++ b/dpctl/tensor/_array_api.py @@ -75,7 +75,7 @@ def __init__(self): self._capabilities = { "boolean indexing": True, "data-dependent shapes": True, - "max dimensions": 64, + "max dimensions": None, } self._all_dtypes = { "bool": dpt.bool, @@ -108,7 +108,7 @@ def capabilities(self): Value: ``True`` ``max dimensions``: integer indication the maximum array dimension supported by ``dpctl``. - Value: ``64`` + Value: ``None`` Returns: dict: diff --git a/dpctl/tests/test_tensor_array_api_inspection.py b/dpctl/tests/test_tensor_array_api_inspection.py index a298517652..bd6d14a528 100644 --- a/dpctl/tests/test_tensor_array_api_inspection.py +++ b/dpctl/tests/test_tensor_array_api_inspection.py @@ -74,7 +74,7 @@ def test_array_api_inspection_capabilities(): capabilities = dpt.__array_namespace_info__().capabilities() assert capabilities["boolean indexing"] assert capabilities["data-dependent shapes"] - assert capabilities["max dimensions"] == 64 + assert capabilities["max dimensions"] is None def test_array_api_inspection_default_dtypes():