From 8b3d6c9f63dbb89cb4733c990bf42a35589884fb Mon Sep 17 00:00:00 2001 From: "Maronas, Marcos" Date: Tue, 23 Jan 2024 04:15:49 -0800 Subject: [PATCH] Fix test in Windows. Signed-off-by: Maronas, Marcos --- sycl/test-e2e/CompositeDevice/composite_device.cpp | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/sycl/test-e2e/CompositeDevice/composite_device.cpp b/sycl/test-e2e/CompositeDevice/composite_device.cpp index 942fd44634c06..0388ffdfd59e5 100644 --- a/sycl/test-e2e/CompositeDevice/composite_device.cpp +++ b/sycl/test-e2e/CompositeDevice/composite_device.cpp @@ -14,8 +14,18 @@ bool isL0Backend(sycl::backend backend) { } bool isCombinedMode() { - const char *Mode = std::getenv("ZE_FLAT_DEVICE_HIERARCHY"); - return (Mode != nullptr) && (std::strcmp(Mode, "COMBINED") == 0); + char *Mode = nullptr; + bool Res = false; +#ifdef _WIN32 + size_t Size = 0; + auto Err = _dupenv_s(&Mode, &Size, "ZE_FLAT_DEVICE_HIERARCHY"); + Res = (Mode != nullptr) && (std::strcmp(Mode, "COMBINED") == 0); + free(Mode); +#else + Mode = std::getenv("ZE_FLAT_DEVICE_HIERARCHY"); + Res = (Mode != nullptr) && (std::strcmp(Mode, "COMBINED") == 0); +#endif + return Res; } int main() {