Merge pull request #1874 from ndellingwood/fix-compatibility-kokkos-4.0

Backward-compatible fix with kokkos@4.0
kokkos · Jun 16, 2023 · 6bbbe6a · 6bbbe6a
2 parents e3a639b + c722dc0
commit 6bbbe6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/batched/dense/impl/KokkosBatched_HostLevel_Gemm_Impl.hpp b/batched/dense/impl/KokkosBatched_HostLevel_Gemm_Impl.hpp
@@ -94,9 +94,11 @@ int BatchedGemmImpl(BatchedGemmHandleType *const handle, const ScalarType alpha,
       case BaseKokkosBatchedAlgos::KK_SERIAL:
       case BaseHeuristicAlgos::SQUARE:
       case BaseTplAlgos::ARMPL:
+#if KOKKOS_VERSION > 40099
         assert(A.rank_dynamic() == 3 && "AViewType must have rank 3.");
         assert(B.rank_dynamic() == 3 && "BViewType must have rank 3.");
         assert(C.rank_dynamic() == 3 && "CViewType must have rank 3.");
+#endif
         break;
       default:
         std::ostringstream os;