Skip to content

Commit 4df56ae

Browse files
author
Francesco Rizzi
committed
blas2: update kokkos include header and cmakelist for tests
1 parent b057ee7 commit 4df56ae

File tree

2 files changed

+75
-0
lines changed

2 files changed

+75
-0
lines changed

tests/kokkos-based/CMakeLists.txt

Lines changed: 73 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,3 +106,76 @@ linalg_add_test_kokkos(
106106
linalg_add_test_kokkos(
107107
copy_kokkos
108108
"copy: kokkos impl")
109+
110+
#
111+
# blas2 (according to P1673)
112+
#
113+
114+
linalg_add_test_kokkos(
115+
overwriting_matrix_vector_product
116+
"overwriting_matrix_vector_product: kokkos impl")
117+
linalg_add_test_kokkos(
118+
updating_matrix_vector_product
119+
"updating_matrix_vector_product: kokkos impl")
120+
121+
linalg_add_test_kokkos(
122+
overwriting_symmetric_matrix_vector_product
123+
"overwriting_symmetric_matrix_vector_product_lower: kokkos impl" USE_LOWER lower)
124+
linalg_add_test_kokkos(
125+
overwriting_symmetric_matrix_vector_product
126+
"overwriting_symmetric_matrix_vector_product_upper: kokkos impl" USE_UPPER upper)
127+
linalg_add_test_kokkos(
128+
updating_symmetric_matrix_vector_product
129+
"updating_symmetric_matrix_vector_product_lower: kokkos impl" USE_LOWER lower)
130+
linalg_add_test_kokkos(
131+
updating_symmetric_matrix_vector_product
132+
"updating_symmetric_matrix_vector_product_upper: kokkos impl" USE_UPPER upper)
133+
134+
linalg_add_test_kokkos(
135+
overwriting_hermitian_matrix_vector_product
136+
"overwriting_hermitian_matrix_vector_product_lower: kokkos impl" USE_LOWER lower)
137+
linalg_add_test_kokkos(
138+
overwriting_hermitian_matrix_vector_product
139+
"overwriting_hermitian_matrix_vector_product_upper: kokkos impl" USE_UPPER upper)
140+
linalg_add_test_kokkos(
141+
updating_hermitian_matrix_vector_product
142+
"updating_hermitian_matrix_vector_product_lower: kokkos impl" USE_LOWER lower)
143+
linalg_add_test_kokkos(
144+
updating_hermitian_matrix_vector_product
145+
"updating_hermitian_matrix_vector_product_upper: kokkos impl" USE_UPPER upper)
146+
147+
148+
linalg_add_test_kokkos(
149+
overwriting_triangular_matrix_vector_product
150+
"overwriting_triangular_matrix_vector_product_lower: kokkos impl" USE_LOWER lower)
151+
linalg_add_test_kokkos(
152+
overwriting_triangular_matrix_vector_product
153+
"overwriting_triangular_matrix_vector_product_upper: kokkos impl" USE_UPPER upper)
154+
linalg_add_test_kokkos(
155+
updating_triangular_matrix_vector_product
156+
"updating_triangular_matrix_vector_product_lower: kokkos impl" USE_LOWER lower)
157+
linalg_add_test_kokkos(
158+
updating_triangular_matrix_vector_product
159+
"updating_triangular_matrix_vector_product_upper: kokkos impl" USE_UPPER upper)
160+
161+
162+
# add tests for triangular_matrix_vector_solve
163+
164+
# FRIZI: leaving these commented out because they don't compile for me
165+
# so Mikolaj is going to finalize them
166+
# linalg_add_test_kokkos(
167+
# matrix_rank1_update_kokkos
168+
# "matrix_rank1_update_kokkos: kokkos impl")
169+
# linalg_add_test_kokkos(
170+
# symmetric_matrix_rank1_update_kokkos
171+
# "symmetric_matrix_rank1_update_kokkos: kokkos impl")
172+
# linalg_add_test_kokkos(
173+
# hermitian_matrix_rank1_update_kokkos
174+
# "hermitian_matrix_rank1_update_kokkos: kokkos impl")
175+
176+
# linalg_add_test_kokkos(
177+
# symmetric_matrix_rank2_update_kokkos
178+
# "symmetric_matrix_rank2_update_kokkos: kokkos impl")
179+
# linalg_add_test_kokkos(
180+
# hermitian_matrix_rank2_update_kokkos
181+
# "hermitian_matrix_rank2_update_kokkos: kokkos impl")

tpl-implementations/include/experimental/linalg_kokkoskernels

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,3 +23,5 @@
2323
#include "__p1673_bits/kokkos-kernels/blas2_matrix_rank_2_update.hpp"
2424
#include "__p1673_bits/kokkos-kernels/blas2_gemv_kk.hpp"
2525
#include "__p1673_bits/kokkos-kernels/blas2_symv_kk.hpp"
26+
#include "__p1673_bits/kokkos-kernels/blas2_hemv_kk.hpp"
27+
#include "__p1673_bits/kokkos-kernels/blas2_triangular_mat_vec_product.hpp"

0 commit comments

Comments
 (0)