We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2450abd commit 1ce6fd2Copy full SHA for 1ce6fd2
batched/dense/impl/KokkosBatched_Pbtrs_Serial_Impl.hpp
@@ -35,6 +35,15 @@ KOKKOS_INLINE_FUNCTION static int checkPbtrsInput([[maybe_unused]] const AViewTy
35
#if (KOKKOSKERNELS_DEBUG_LEVEL > 0)
36
const int ldb = x.extent(0);
37
const int lda = A.extent(0), n = A.extent(1);
38
+ const int kd = lda - 1;
39
+ if (kd < 0) {
40
+ Kokkos::printf(
41
+ "KokkosBatched::pbtrs: leading dimension of A must not be less than 1: %d, A: "
42
+ "%d "
43
+ "x %d \n",
44
+ lda, n);
45
+ return 1;
46
+ }
47
if (ldb < Kokkos::max(1, n)) {
48
Kokkos::printf(
49
"KokkosBatched::pbtrs: Dimensions of x and A do not match: x: %d, A: "
0 commit comments