From c722dc0a5e1bb4390eff83d9ef72cd3312ac7156 Mon Sep 17 00:00:00 2001
From: Nathan Ellingwood <ndellin@sandia.gov>
Date: Fri, 16 Jun 2023 14:07:39 -0600
Subject: [PATCH] Backward-compatible fix with kokkos@4.0

---
 batched/dense/impl/KokkosBatched_HostLevel_Gemm_Impl.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/batched/dense/impl/KokkosBatched_HostLevel_Gemm_Impl.hpp b/batched/dense/impl/KokkosBatched_HostLevel_Gemm_Impl.hpp
index 7a93309e65..f413ba612c 100644
--- 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;