Skip to content

Commit

Permalink
Batched - QR: fix execution space in unit-test views
Browse files Browse the repository at this point in the history
Signed-off-by: Luc <[email protected]>
  • Loading branch information
lucbv committed Oct 8, 2024
1 parent 46fdfad commit 95d1181
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions batched/dense/unit_test/Test_Batched_SerialQR.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -299,10 +299,10 @@ void test_QR_batch() {

constexpr int numMat = 314;
constexpr int maxMatSize = 36;
Kokkos::View<Scalar**> tau("tau", numMat, maxMatSize);
Kokkos::View<Scalar**> tmp("work buffer", numMat, maxMatSize);
Kokkos::View<Scalar***> mats("matrices", numMat, maxMatSize, maxMatSize);
Kokkos::View<Scalar***> Qs("Q matrices", numMat, maxMatSize, maxMatSize);
Kokkos::View<Scalar**, ExecutionSpace> tau("tau", numMat, maxMatSize);
Kokkos::View<Scalar**, ExecutionSpace> tmp("work buffer", numMat, maxMatSize);
Kokkos::View<Scalar***, ExecutionSpace> mats("matrices", numMat, maxMatSize, maxMatSize);
Kokkos::View<Scalar***, ExecutionSpace> Qs("Q matrices", numMat, maxMatSize, maxMatSize);

Kokkos::Random_XorShift64_Pool<ExecutionSpace> rand_pool(2718);
constexpr double max_val = 1000;
Expand Down

0 comments on commit 95d1181

Please sign in to comment.