oneapi sparse (arrayfire#3469)

* adds sparse_arith, sparse blas to oneapi backend
WilliamTambellini · Jul 8, 2023 · d74656f · d74656f
1 parent e9132c2
commit d74656f
Show file tree

Hide file tree

Showing 7 changed files with 1,308 additions and 259 deletions.
diff --git a/src/backend/common/Binary.hpp b/src/backend/common/Binary.hpp
@@ -40,13 +40,27 @@ struct Binary<T, af_add_t> {
     __DH__ T operator()(T lhs, T rhs) { return lhs + rhs; }
 };
 
+template<typename T>
+struct Binary<T, af_sub_t> {
+    static __DH__ T init() { return scalar<T>(0); }
+
+    __DH__ T operator()(T lhs, T rhs) { return lhs - rhs; }
+};
+
 template<typename T>
 struct Binary<T, af_mul_t> {
     static __DH__ T init() { return scalar<T>(1); }
 
     __DH__ T operator()(T lhs, T rhs) { return lhs * rhs; }
 };
 
+template<typename T>
+struct Binary<T, af_div_t> {
+    static __DH__ T init() { return scalar<T>(1); }
+
+    __DH__ T operator()(T lhs, T rhs) { return lhs / rhs; }
+};
+
 template<typename T>
 struct Binary<T, af_or_t> {
     static __DH__ T init() { return scalar<T>(0); }

diff --git a/src/backend/oneapi/CMakeLists.txt b/src/backend/oneapi/CMakeLists.txt
@@ -249,6 +249,8 @@ target_sources(afoneapi
     kernel/scan_dim.hpp
     kernel/sort.hpp
     kernel/sort_by_key.hpp
+    kernel/sparse.hpp
+    kernel/sparse_arith.hpp
     kernel/transpose.hpp
     kernel/transpose_inplace.hpp
     kernel/triangle.hpp