Upgrade finch-tensor (#694)

* Upgrade finch-tensor

* Change to_device to to_storage

* Change to_device to to_storage

Author: mtsokol

benchmarks/benchmark_backends.py

@@ -24,10 +24,10 @@ def setup(self):
         if sparse.BackendType.Finch == BACKEND:
             import finch
 
-            self.s1 = self.s1.to_device(
+            self.s1 = self.s1.to_storage(
                 finch.Storage(finch.Dense(finch.SparseList(finch.Element(0.0))), order=self.s1.get_order())
             )
-            self.s2 = self.s2.to_device(
+            self.s2 = self.s2.to_storage(
                 finch.Storage(
                     finch.Dense(finch.SparseList(finch.SparseList(finch.Element(0.0)))),
                     order=self.s2.get_order(),
@@ -60,7 +60,7 @@ def setup(self, lazy_mode, size_and_density):
         if sparse.BackendType.Finch == BACKEND:
             import finch
 
-            self.M = self.M.to_device(
+            self.M = self.M.to_storage(
                 finch.Storage(finch.Dense(finch.SparseList(finch.Element(0.0))), order=self.M.get_order())
             )
             self.v1 = finch.Tensor(self.v1)
@@ -108,10 +108,10 @@ def setup(self):
         if sparse.BackendType.Finch == BACKEND:
             import finch
 
-            self.s1 = self.s1.to_device(
+            self.s1 = self.s1.to_storage(
                 finch.Storage(finch.Dense(finch.SparseList(finch.Element(0.0))), order=self.s1.get_order())
             )
-            self.s2 = self.s2.to_device(
+            self.s2 = self.s2.to_storage(
                 finch.Storage(
                     finch.Dense(finch.SparseList(finch.SparseList(finch.Element(0.0)))), order=self.s2.get_order()
                 )

ci/Finch-array-api-xfails.txt

@@ -18,8 +18,6 @@ array_api_tests/test_signatures.py::test_func_signature[linspace]
 array_api_tests/test_signatures.py::test_func_signature[meshgrid]
 array_api_tests/test_signatures.py::test_func_signature[tril]
 array_api_tests/test_signatures.py::test_func_signature[triu]
-# missing 'copy' keyword
-array_api_tests/test_signatures.py::test_func_signature[asarray]
 # inspection functions
 array_api_tests/test_signatures.py::test_func_signature[isdtype]
 array_api_tests/test_signatures.py::test_func_signature[result_type]
@@ -70,8 +68,6 @@ array_api_tests/test_signatures.py::test_extension_func_signature[linalg.vector_
 array_api_tests/test_signatures.py::test_array_method_signature[__dlpack__]
 array_api_tests/test_signatures.py::test_array_method_signature[__dlpack_device__]
 array_api_tests/test_signatures.py::test_array_method_signature[__setitem__]
-# missing 'stream' from signature
-array_api_tests/test_signatures.py::test_array_method_signature[to_device]
 
 # test_creation_functions
 
@@ -91,37 +87,10 @@ array_api_tests/test_array_object.py::test_setitem_masking
 
 # test_operators_and_elementwise_functions
 
-# inplace operators not supported
-array_api_tests/test_operators_and_elementwise_functions.py::test_add[__iadd__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_add[__iadd__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_and[__iand__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_and[__iand__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_left_shift[__ilshift__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_left_shift[__ilshift__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_or[__ior__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_or[__ior__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_right_shift[__irshift__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_right_shift[__irshift__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_xor[__ixor__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_bitwise_xor[__ixor__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_divide[__itruediv__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_divide[__itruediv__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[__ifloordiv__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[__ifloordiv__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_multiply[__imul__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_multiply[__imul__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_pow[__ipow__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_pow[__ipow__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_subtract[__isub__(x1, x2)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_subtract[__isub__(x, s)]
-# type promotion differences
-array_api_tests/test_operators_and_elementwise_functions.py::test_add[__add__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_divide[__truediv__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_multiply[__mul__(x, s)]
-array_api_tests/test_operators_and_elementwise_functions.py::test_subtract[__sub__(x, s)]
 # ArgumentError: invalid rational: zero(...)//zero(...)
 array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[floor_divide(x1, x2)]
 array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[__floordiv__(x1, x2)]
+array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[__ifloordiv__(x1, x2)]
 # throws for x < 1 instead of NaN
 array_api_tests/test_operators_and_elementwise_functions.py::test_acosh
 # not implemented
@@ -211,7 +180,6 @@ array_api_tests/test_has_names.py::test_has_names[array_method-__dlpack__]
 array_api_tests/test_has_names.py::test_has_names[array_method-__dlpack_device__]
 array_api_tests/test_has_names.py::test_has_names[array_method-__setitem__]
 array_api_tests/test_has_names.py::test_has_names[array_attribute-T]
-array_api_tests/test_has_names.py::test_has_names[array_attribute-device]
 array_api_tests/test_has_names.py::test_has_names[array_attribute-mT]
 
 # test_indexing_functions

pyproject.toml

@@ -38,7 +38,7 @@ tests = [
 ]
 tox = ["sparse[tests]", "tox"]
 all = ["sparse[docs,tox]", "matrepr"]
-finch = ["finch-tensor>=0.1.25"]
+finch = ["finch-tensor>=0.1.26"]
 
 [project.urls]
 Documentation = "https://sparse.pydata.org/"

sparse/tests/test_backends.py

@@ -21,8 +21,8 @@ def test_backend_contex_manager(backend):
         def storage():
             return finch.Storage(finch.Dense(finch.SparseList(finch.SparseList(finch.Element(0.0)))), order="C")
 
-        x = x.to_device(storage())
-        y = y.to_device(storage())
+        x = x.to_storage(storage())
+        y = y.to_storage(storage())
     else:
         x.asformat("gcxs")
         y.asformat("gcxs")