SWDEV-543340 - Added Unit_hipEventIpc_shm_cleanup test (#548)

The test verifies that all shared memory objects for
IPC events used internally by HIP are properly cleaned
up after use and do not leave persistent files in /dev/shm.

Co-authored-by: Ioannis Assiouras <Ioannis.Assiouras@amd.com>
Этот коммит содержится в:
systems-assistant[bot]
2025-11-13 18:21:12 +00:00
коммит произвёл GitHub
родитель a0f53a5fdb
Коммит f55dda2082
2 изменённых файлов: 148 добавлений и 0 удалений
+5
Просмотреть файл
@@ -17,6 +17,11 @@ if(HIP_PLATFORM MATCHES "amd")
set(AMD_SRC
Unit_hipEventQuery.cc
)
if(UNIX)
set(AMD_SRC
${AMD_SRC}
Unit_hipEventIpc_shm_cleanup.cc)
endif()
set(TEST_SRC ${TEST_SRC} ${AMD_SRC})
endif()
+143
Просмотреть файл
@@ -0,0 +1,143 @@
/*
Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
*/
// Test hipEventIpc behavior.
#include <hip_test_checkers.hh>
#include <hip_test_kernels.hh>
#include <hip_test_common.hh>
#include <dirent.h>
#include <string>
#include <set>
#include <regex>
#include <sys/wait.h>
#include <unistd.h>
/**
* Test Description
* ------------------------
* - Verify that all shared memory objects for IPC events used internally by HIP are properly
* cleaned up after use and do not leave persistent files in /dev/shm.
* Test source
* ------------------------
* - unit/event/Unit_hipEventIpc_shm_cleanup.cc
* Test requirements
* ------------------------
* - HIP_VERSION >= 7.0
*/
// Helper to get current /dev/shm files that look like HIP IPC
std::set<std::string> get_hip_ipc_shm_files() {
std::set<std::string> result;
DIR* dir = opendir("/dev/shm");
if (!dir) return result;
struct dirent* entry;
std::regex hip_pattern("^hip_.*"); // Matches typical HIP shm names
while ((entry = readdir(dir)) != nullptr) {
if (std::regex_match(entry->d_name, hip_pattern)) {
result.insert(entry->d_name);
}
}
closedir(dir);
return result;
}
TEST_CASE("Unit_hipEventIpc_shm_cleanup") {
auto before = get_hip_ipc_shm_files();
int pipefd[2];
REQUIRE(pipe(pipefd) == 0); // Create pipe: pipefd[0] for reading, pipefd[1] for writing
pid_t pid = fork();
REQUIRE(pid >= 0); // Ensure fork succeeded
if (pid == 0) {
// === Child Process ===
close(pipefd[1]); // Close write end
hipIpcEventHandle_t handle;
ssize_t total = 0;
char* ptr = reinterpret_cast<char*>(&handle);
while (total < sizeof(handle)) {
ssize_t r = read(pipefd[0], ptr + total, sizeof(handle) - total);
if (r <= 0) {
perror("Child failed to read from pipe");
exit(1);
}
total += r;
}
close(pipefd[0]);
hipEvent_t event;
if (hipIpcOpenEventHandle(&event, handle) != hipSuccess) {
perror("Child failed to open event handle");
exit(1);
}
if (hipEventDestroy(event) != hipSuccess) {
perror("Child failed to destroy event");
exit(1);
}
exit(0);
} else {
// === Parent Process ===
close(pipefd[0]); // Close read end
hipEvent_t event;
REQUIRE(hipEventCreateWithFlags(&event, hipEventInterprocess | hipEventDisableTiming) ==
hipSuccess);
hipIpcEventHandle_t ipcHandle;
REQUIRE(hipIpcGetEventHandle(&ipcHandle, event) == hipSuccess);
// Write handle to child
ssize_t total = 0;
const char* ptr = reinterpret_cast<const char*>(&ipcHandle);
while (total < sizeof(ipcHandle)) {
ssize_t w = write(pipefd[1], ptr + total, sizeof(ipcHandle) - total);
if (w <= 0) {
perror("Parent failed to write to pipe");
exit(1);
}
total += w;
}
close(pipefd[1]);
// Wait for child
int status;
waitpid(pid, &status, 0);
REQUIRE(WIFEXITED(status));
REQUIRE(WEXITSTATUS(status) == 0);
REQUIRE(hipEventDestroy(event) == hipSuccess);
auto after = get_hip_ipc_shm_files();
std::cout << before.size() << after.size() << std::endl;
for (const auto& file : after) {
REQUIRE(before.count(file) == 1); // No new HIP shm files left behind
}
}
}