From 74c4a248cce362bdcad8912b70f1d27e6a27602f Mon Sep 17 00:00:00 2001
From: Brandon Potter <brandon.potter@amd.com>
Date: Mon, 9 Sep 2024 10:14:06 -0700
Subject: [PATCH] Add an extra assertion check for nullptr

[ROCm/rocshmem commit: 678564ba3ce1d772745b16c9d1c4b2908a7351c0]
---
 .../rocshmem/tests/unit_tests/ipc_impl_simple_coarse_gtest.cpp   | 1 +
 .../rocshmem/tests/unit_tests/ipc_impl_simple_fine_gtest.cpp     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/projects/rocshmem/tests/unit_tests/ipc_impl_simple_coarse_gtest.cpp b/projects/rocshmem/tests/unit_tests/ipc_impl_simple_coarse_gtest.cpp
index dbf72923b5..37833f9cfe 100644
--- a/projects/rocshmem/tests/unit_tests/ipc_impl_simple_coarse_gtest.cpp
+++ b/projects/rocshmem/tests/unit_tests/ipc_impl_simple_coarse_gtest.cpp
@@ -33,6 +33,7 @@ TEST_F(IPCImplSimpleCoarseTestFixture, MPI_num_pes) {
 }
 
 TEST_F(IPCImplSimpleCoarseTestFixture, IPC_bases) {
+  ASSERT_NE(ipc_impl_.ipc_bases, nullptr);
   for(int i{0}; i < mpi_.num_pes(); i++) {
     ASSERT_NE(ipc_impl_.ipc_bases[i], nullptr);
   }
diff --git a/projects/rocshmem/tests/unit_tests/ipc_impl_simple_fine_gtest.cpp b/projects/rocshmem/tests/unit_tests/ipc_impl_simple_fine_gtest.cpp
index 1c4c409f59..7c752e2f6a 100644
--- a/projects/rocshmem/tests/unit_tests/ipc_impl_simple_fine_gtest.cpp
+++ b/projects/rocshmem/tests/unit_tests/ipc_impl_simple_fine_gtest.cpp
@@ -33,6 +33,7 @@ TEST_F(IPCImplSimpleFineTestFixture, MPI_num_pes) {
 }
 
 TEST_F(IPCImplSimpleFineTestFixture, IPC_bases) {
+  ASSERT_NE(ipc_impl_.ipc_bases, nullptr);
   for(int i{0}; i < mpi_.num_pes(); i++) {
     ASSERT_NE(ipc_impl_.ipc_bases[i], nullptr);
   }