Upgrade finch-tensor (#702)

* Upgrade finch-tensor

* Bump finch-tensor version

---------

Co-authored-by: Hameer Abbasi <[email protected]>

Author: mtsokol

ci/Finch-array-api-xfails.txt

@@ -11,10 +11,6 @@ array_api_tests/test_signatures.py::test_func_signature[unique_counts]
 array_api_tests/test_signatures.py::test_func_signature[unique_inverse]
 array_api_tests/test_signatures.py::test_func_signature[unique_values]
 # creation functions
-array_api_tests/test_signatures.py::test_func_signature[empty]
-array_api_tests/test_signatures.py::test_func_signature[empty_like]
-array_api_tests/test_signatures.py::test_func_signature[arange]
-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]
@@ -72,10 +68,6 @@ array_api_tests/test_signatures.py::test_array_method_signature[__setitem__]
 # test_creation_functions
 
 # not implemented
-array_api_tests/test_creation_functions.py::test_arange
-array_api_tests/test_creation_functions.py::test_empty
-array_api_tests/test_creation_functions.py::test_empty_like
-array_api_tests/test_creation_functions.py::test_linspace
 array_api_tests/test_creation_functions.py::test_meshgrid
 
 # test_array_object
@@ -87,10 +79,6 @@ array_api_tests/test_array_object.py::test_setitem_masking
 
 # test_operators_and_elementwise_functions
 
-# 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
@@ -142,11 +130,7 @@ array_api_tests/test_has_names.py::test_has_names[set-unique_values]
 array_api_tests/test_has_names.py::test_has_names[searching-argmax]
 array_api_tests/test_has_names.py::test_has_names[searching-argmin]
 array_api_tests/test_has_names.py::test_has_names[searching-searchsorted]
-array_api_tests/test_has_names.py::test_has_names[creation-arange]
-array_api_tests/test_has_names.py::test_has_names[creation-empty]
-array_api_tests/test_has_names.py::test_has_names[creation-empty_like]
 array_api_tests/test_has_names.py::test_has_names[creation-from_dlpack]
-array_api_tests/test_has_names.py::test_has_names[creation-linspace]
 array_api_tests/test_has_names.py::test_has_names[creation-meshgrid]
 array_api_tests/test_has_names.py::test_has_names[creation-tril]
 array_api_tests/test_has_names.py::test_has_names[creation-triu]

examples/matmul_example.py

@@ -31,7 +31,7 @@
 
     @sparse.compiled
     def sddmm_finch(a, b):
-        return sparse.sum(a[:, None, :] * sparse.permute_dims(b, (1, 0))[None, :, :], axis=-1)
+        return a @ b
 
     # Compile
     result_finch = sddmm_finch(a, b)

pyproject.toml

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

sparse/finch_backend/__init__.py

@@ -11,6 +11,7 @@
     add,
     all,
     any,
+    arange,
     asarray,
     asin,
     asinh,
@@ -36,6 +37,8 @@
     cosh,
     divide,
     e,
+    empty,
+    empty_like,
     equal,
     exp,
     expm1,
@@ -64,6 +67,7 @@
     lazy,
     less,
     less_equal,
+    linspace,
     log,
     log1p,
     log2,
@@ -223,4 +227,8 @@
     "real",
     "imag",
     "conj",
+    "empty",
+    "empty_like",
+    "arange",
+    "linspace",
 ]