forked from kokkos/stdBLAS
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlinalg_kokkoskernels
More file actions
34 lines (31 loc) · 1.76 KB
/
linalg_kokkoskernels
File metadata and controls
34 lines (31 loc) · 1.76 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#pragma once
#include<experimental/mdspan>
#include<KokkosBlas.hpp>
#include "__p1673_bits/kokkos-kernels/mdspan_to_view_mapper_kk.hpp"
#include "__p1673_bits/kokkos-kernels/kokkos_conjugate.hpp"
// blas1 (according to P1673)
#include "__p1673_bits/kokkos-kernels/blas1_dot_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_add_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_scale_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_idx_abs_max_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_vector_norm2_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_vector_abs_sum_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_vector_sum_of_squares_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_matrix_frob_norm_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_matrix_inf_norm_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_matrix_one_norm_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_swap_elements_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas1_copy_kk.hpp"
// blas2 (according to P1673)
#include "__p1673_bits/kokkos-kernels/blas2_matrix_rank_1_update.hpp"
#include "__p1673_bits/kokkos-kernels/blas2_matrix_rank_2_update.hpp"
#include "__p1673_bits/kokkos-kernels/blas2_gemv_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas2_symv_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas2_hemv_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas2_triangular_mat_vec_product.hpp"
// blas3 (according to P1673)
#include "__p1673_bits/kokkos-kernels/blas3_overwriting_gemm_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas3_matrix_rank_k_update.hpp"
#include "__p1673_bits/kokkos-kernels/blas3_matrix_rank_2k_update.hpp"
#include "__p1673_bits/kokkos-kernels/blas3_matrix_product_kk.hpp"
#include "__p1673_bits/kokkos-kernels/blas3_triangular_matrix_matrix_solve.hpp"