From 5525a24c7a1e71464c57d9ab940bb203d079b716 Mon Sep 17 00:00:00 2001 From: Rafal Rudnicki Date: Mon, 15 Sep 2025 14:18:32 +0000 Subject: [PATCH] x --- test/providers/provider_cuda.cpp | 28 ++++++++++++++++++++-------- 1 file changed, 20 insertions(+), 8 deletions(-) diff --git a/test/providers/provider_cuda.cpp b/test/providers/provider_cuda.cpp index 8a3d729eb..db466d152 100644 --- a/test/providers/provider_cuda.cpp +++ b/test/providers/provider_cuda.cpp @@ -802,12 +802,24 @@ INSTANTIATE_TEST_SUITE_P( return names[info.index]; })); -// TODO: add IPC API +CUDATestHelper cuTestHelper; + +CUDAMemoryAccessor cuAccessor(cuTestHelper.get_test_context(), + cuTestHelper.get_test_device()); + +void *createCuParamsDeviceMemory() { + return create_cuda_prov_params(cuTestHelper.get_test_context(), + cuTestHelper.get_test_device(), + UMF_MEMORY_TYPE_DEVICE, 0); +} + +#ifdef _WIN32 GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(umfIpcTest); -/* -INSTANTIATE_TEST_SUITE_P(umfCUDAProviderTestSuite, umfIpcTest, - ::testing::Values(ipcTestParams{ - umfProxyPoolOps(), nullptr, - umfCUDAMemoryProviderOps(), - cuParams_device_memory.get(), &cuAccessor, false})); -*/ +#else +INSTANTIATE_TEST_SUITE_P( + umfCUDAProviderTestSuite, umfIpcTest, + ::testing::Values(ipcTestParams{ + umfProxyPoolOps(), nullptr, nullptr, umfCUDAMemoryProviderOps(), + createCUDAParamsDeviceMemory, destroyCuParams, &l0Accessor}), + ipcTestParamsNameGen); +#endif