@@ -25,20 +25,24 @@ namespace KokkosBatched {
25
25
// / \brief Serial Batched Gbtrf:
26
26
// / Compute a LU factorization of a m-by-n band matrix A using partial
27
27
// / pivoting with row interchanges.
28
+ // / \tparam ArgAlgo: Type indicating the blocked (KokkosBatched::Algo::Gbtrf::Blocked) or unblocked
29
+ // / (KokkosBatched::Algo::Gbtrf::Unblocked) algorithm to be used
30
+ // /
28
31
// / \tparam ABViewType: Input type for the matrix, needs to be a 2D view
29
32
// / \tparam PivViewType: Input type for the pivot indices, needs to be a 1D view
30
33
// /
31
34
// / \param Ab [inout]: A is a ldab by n band matrix, a rank 2 view
32
35
// / \param piv [out]: On exit, the pivot indices, a rank 1 view
33
36
// / \param kl [in]: The number of subdiagonals within the band of A. kl >= 0
34
37
// / \param ku [in]: The number of superdiagonals within the band of A. ku >= 0
35
- // / \param m [in]: The number of rows of the matrix A. (optional)
38
+ // / \param m [in]: The number of rows of the matrix A. (optional, default is -1, corresponding to m == n )
36
39
// /
37
40
// / No nested parallel_for is used inside of the function.
38
41
// /
39
42
40
43
template <typename ArgAlgo>
41
44
struct SerialGbtrf {
45
+ static_assert (std::is_same_v<ArgAlgo, Algo::Gbtrf::Unblocked>, " KokkosBatched::gbtrf: Use Algo::Gbtrf::Unblocked" );
42
46
template <typename ABViewType, typename PivViewType>
43
47
KOKKOS_INLINE_FUNCTION static int invoke (const ABViewType &Ab, const PivViewType &piv, const int kl, const int ku,
44
48
const int m = -1 );
0 commit comments