From 87d2efa4305237cb1f323fd5d11e5a63d14b0295 Mon Sep 17 00:00:00 2001
From: Aurelien Bouteiller <Aurelien.bouteiller@gmail.com>
Date: Thu, 12 Jun 2025 13:16:30 -0400
Subject: [PATCH] SWDEV-536571 - Include assert header. (#157)

(cherry picked from commit 96cd36e47e4759b7515373e8fa455385f3d985aa)

Co-authored-by: Patel <jaypatel@amd.com>

[ROCm/rocshmem commit: bcc14b1a34eecd62d458a15f0c2b901fbe254f48]
---
 projects/rocshmem/src/containers/index_strategy.hpp | 2 +-
 projects/rocshmem/src/team_tracker.cpp              | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/projects/rocshmem/src/containers/index_strategy.hpp b/projects/rocshmem/src/containers/index_strategy.hpp
index 7609fff216..682c7821d6 100644
--- a/projects/rocshmem/src/containers/index_strategy.hpp
+++ b/projects/rocshmem/src/containers/index_strategy.hpp
@@ -132,7 +132,7 @@
  */
 
 #include <hip/hip_runtime.h>
-
+#include <cassert>
 #include <algorithm>
 
 namespace rocshmem {
diff --git a/projects/rocshmem/src/team_tracker.cpp b/projects/rocshmem/src/team_tracker.cpp
index 119e2635dc..343b414164 100644
--- a/projects/rocshmem/src/team_tracker.cpp
+++ b/projects/rocshmem/src/team_tracker.cpp
@@ -23,7 +23,7 @@
  *****************************************************************************/
 
 #include "team_tracker.hpp"
-
+#include <cassert>
 #include <cstdlib>
 
 namespace rocshmem {