SWDEV-311270 - Fix HIP tests for mempool IPC

Change-Id: Ib5065a259e66fd53c3ffd6bc038dd65be00a3a3d
Этот коммит содержится в:
German Andryeyev
2024-02-16 16:37:42 -05:00
родитель 978e165d33
Коммит 228748a9d8
2 изменённых файлов: 8 добавлений и 3 удалений
+7 -2
Просмотреть файл
@@ -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));
+1 -1
Просмотреть файл
@@ -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));