SWDEV-311270 - Fix HIP tests for mempool IPC

Change-Id: Ib5065a259e66fd53c3ffd6bc038dd65be00a3a3d
This commit is contained in:
German Andryeyev committed 2024-02-20 11:35:44 -05:00
1 parent 978e165d33
commit 228748a9d8
2 files changed
+8 -3

No files matched your search

@@ -29,14 +29,19 @@ class MemPoolImportFromShareableHandleBenchmark : public Benchmark<MemPoolImport
public:
void operator()() {
hipMemPool_t mem_pool{nullptr};
int share_handle;
#if defined(_WIN32)
void* share_handle;
#else
int64_t share_handle;
#endif
hipMemPoolProps props = CreateMemPoolProps(0, kHandleType);
HIP_CHECK(hipMemPoolCreate(&mem_pool, &props));
HIP_CHECK(hipMemPoolExportToShareableHandle(&share_handle, mem_pool, kHandleType, 0));
TIMED_SECTION(kTimerTypeCpu) {
HIP_CHECK(hipMemPoolImportFromShareableHandle(&mem_pool, &share_handle, kHandleType, 0));
HIP_CHECK(hipMemPoolImportFromShareableHandle(
&mem_pool, (void*)share_handle, kHandleType, 0));
}
HIP_CHECK(hipMemPoolDestroy(mem_pool));
@@ -40,7 +40,7 @@ class MemPoolImportPointerBenchmark : public Benchmark<MemPoolImportPointerBench
HIP_CHECK(hipMemPoolExportPointer(&exp_data, device_ptr));
TIMED_SECTION(kTimerTypeCpu) {
HIP_CHECK(hipMemPoolImportPointer(reinterpret_cast<void**>(device_ptr_import), mem_pool, &exp_data));
HIP_CHECK(hipMemPoolImportPointer(reinterpret_cast<void**>(&device_ptr_import), mem_pool, &exp_data));
}
HIP_CHECK(hipFree(device_ptr));