Skip to content

Commit

Permalink
BatchedGemm test: workaround testing cublas+magma
Browse files Browse the repository at this point in the history
- temporary workaround to skip magma test when cublas enabled to avoid issues like kokkos#2177
  • Loading branch information
ndellingwood committed Apr 23, 2024
1 parent 3f4744e commit 0d1f813
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion batched/dense/unit_test/Test_Batched_BatchedGemm.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,11 @@ void impl_test_batched_gemm(const int N, const int matAdim1, const int matAdim2,
ASSERT_EQ(batchedGemmHandleCublas.vecLen, 0);
#endif

#if defined(KOKKOSKERNELS_ENABLE_TPL_MAGMA)
// FIXME temporary workaround to run this magma test only if cublas is not
// enabled the design of the BatchedGemmHandle currently does not allow
// simultanous testing in this way. See issue #2177
#if defined(KOKKOSKERNELS_ENABLE_TPL_MAGMA) && \
!defined(KOKKOSKERNELS_ENABLE_TPL_CUBLAS)
magma_queue_t magma_queue;
BatchedGemmHandle batchedGemmHandleMagma(magma_queue, GemmTplAlgos::MAGMA,
0, 0);
Expand Down

0 comments on commit 0d1f813

Please sign in to comment.