[Error] Better error message when creating sparse snodes on backends that do not support sparse

python/taichi/_snode/fields_builder.py

@@ -74,9 +74,10 @@ def dense(self, indices: Union[Sequence[_Axis], _Axis],
     def pointer(self, indices: Union[Sequence[_Axis], _Axis],
                 dimensions: Union[Sequence[int], int]):
         """Same as :func:`taichi.lang.snode.SNode.pointer`"""
-        if impl.current_cfg().arch == _ti_core.metal:
+        if not _ti_core.is_extension_supported(impl.current_cfg().arch,
+                                               _ti_core.Extension.sparse):
             raise TaichiRuntimeError(
-                "Pointer SNode on metal backend is deprecated and removed.")
+                "Pointer SNode is not supported on this backend.")
         self._check_not_finalized()
         self.empty = False
         return self.root.pointer(indices, dimensions)
@@ -90,19 +91,21 @@ def dynamic(self,
                 dimension: Union[Sequence[int], int],
                 chunk_size: Optional[int] = None):
         """Same as :func:`taichi.lang.snode.SNode.dynamic`"""
-        if impl.current_cfg().arch == _ti_core.metal:
+        if not _ti_core.is_extension_supported(impl.current_cfg().arch,
+                                               _ti_core.Extension.sparse):
             raise TaichiRuntimeError(
-                "Dynamic SNode on metal backend is deprecated and removed.")
+                "Dynamic SNode is not supported on this backend.")
         self._check_not_finalized()
         self.empty = False
         return self.root.dynamic(index, dimension, chunk_size)
 
     def bitmasked(self, indices: Union[Sequence[_Axis], _Axis],
                   dimensions: Union[Sequence[int], int]):
         """Same as :func:`taichi.lang.snode.SNode.bitmasked`"""
-        if impl.current_cfg().arch == _ti_core.metal:
+        if not _ti_core.is_extension_supported(impl.current_cfg().arch,
+                                               _ti_core.Extension.sparse):
             raise TaichiRuntimeError(
-                "Bitmasked SNode on metal backend is deprecated and removed.")
+                "Bitmasked SNode is not supported on this backend.")
         self._check_not_finalized()
         self.empty = False
         return self.root.bitmasked(indices, dimensions)

python/taichi/lang/snode.py

@@ -46,9 +46,10 @@ def pointer(self, axes, dimensions):
         Returns:
             The added :class:`~taichi.lang.SNode` instance.
         """
-        if impl.current_cfg().arch == _ti_core.metal:
+        if not _ti_core.is_extension_supported(impl.current_cfg().arch,
+                                               _ti_core.Extension.sparse):
             raise TaichiRuntimeError(
-                "Pointer SNode on metal backend is deprecated and removed.")
+                "Pointer SNode is not supported on this backend.")
         if isinstance(dimensions, numbers.Number):
             dimensions = [dimensions] * len(axes)
         return SNode(self.ptr.pointer(axes, dimensions, get_traceback()))
@@ -73,9 +74,10 @@ def dynamic(self, axis, dimension, chunk_size=None):
         Returns:
             The added :class:`~taichi.lang.SNode` instance.
         """
-        if impl.current_cfg().arch == _ti_core.metal:
+        if not _ti_core.is_extension_supported(impl.current_cfg().arch,
+                                               _ti_core.Extension.sparse):
             raise TaichiRuntimeError(
-                "Dynamic SNode on metal backend is deprecated and removed.")
+                "Dynamic SNode is not supported on this backend.")
         assert len(axis) == 1
         if chunk_size is None:
             chunk_size = dimension
@@ -92,9 +94,10 @@ def bitmasked(self, axes, dimensions):
         Returns:
             The added :class:`~taichi.lang.SNode` instance.
         """
-        if impl.current_cfg().arch == _ti_core.metal:
+        if not _ti_core.is_extension_supported(impl.current_cfg().arch,
+                                               _ti_core.Extension.sparse):
             raise TaichiRuntimeError(
-                "Bitmasked SNode on metal backend is deprecated and removed.")
+                "Bitmasked SNode is not supported on this backend.")
         if isinstance(dimensions, numbers.Number):
             dimensions = [dimensions] * len(axes)
         return SNode(self.ptr.bitmasked(axes, dimensions, get_traceback()))

tests/python/test_deprecation.py

@@ -74,24 +74,6 @@ def test_deprecate_element_shape_ndarray_arg():
                      element_shape=(1, ))
 
 
-# Remove this before v1.5.0
-@test_utils.test(arch=ti.metal)
-def test_deprecate_metal_sparse():
-    with pytest.raises(
-            ti.TaichiRuntimeError,
-            match="Pointer SNode on metal backend is deprecated and removed."):
-        ti.root.pointer(ti.i, 10)
-    with pytest.raises(
-            ti.TaichiRuntimeError,
-            match="Bitmasked SNode on metal backend is deprecated and removed."
-    ):
-        ti.root.bitmasked(ti.j, 10)
-    with pytest.raises(
-            ti.TaichiRuntimeError,
-            match="Dynamic SNode on metal backend is deprecated and removed."):
-        ti.root.dynamic(ti.i, 10)
-
-
 @test_utils.test(arch=ti.vulkan)
 def test_deprecated_rwtexture_type():
     n = 128

tests/python/test_field.py

@@ -324,3 +324,39 @@ def foo():
             b = a * x
 
         foo()
+
+
+@test_utils.test(exclude=[ti.x64, ti.arm64, ti.cuda])
+def test_sparse_not_supported():
+    with pytest.raises(
+            ti.TaichiRuntimeError,
+            match="Pointer SNode is not supported on this backend."):
+        ti.root.pointer(ti.i, 10)
+
+    with pytest.raises(
+            ti.TaichiRuntimeError,
+            match="Pointer SNode is not supported on this backend."):
+        a = ti.root.dense(ti.i, 10)
+        a.pointer(ti.j, 10)
+
+    with pytest.raises(
+            ti.TaichiRuntimeError,
+            match="Dynamic SNode is not supported on this backend."):
+        ti.root.dynamic(ti.i, 10)
+
+    with pytest.raises(
+            ti.TaichiRuntimeError,
+            match="Dynamic SNode is not supported on this backend."):
+        a = ti.root.dense(ti.i, 10)
+        a.dynamic(ti.j, 10)
+
+    with pytest.raises(
+            ti.TaichiRuntimeError,
+            match="Bitmasked SNode is not supported on this backend."):
+        ti.root.bitmasked(ti.i, 10)
+
+    with pytest.raises(
+            ti.TaichiRuntimeError,
+            match="Bitmasked SNode is not supported on this backend."):
+        a = ti.root.dense(ti.i, 10)
+        a.bitmasked(ti.j, 10)