diff --git a/projects/hip-tests/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc b/projects/hip-tests/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc index 03fa33636c..ea65bd4ead 100644 --- a/projects/hip-tests/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc +++ b/projects/hip-tests/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc @@ -184,17 +184,15 @@ TEST_CASE("Unit_hipGraphExecMemcpyNodeSetParamsFromSymbol_Negative_Parameters") } #endif - SECTION("Changing dst allocation device") { - if (HipTest::getDeviceCount() < 2) { - HipTest::HIP_SKIP_TEST("Test requires two connected GPUs"); - return; - } + if (HipTest::getDeviceCount() >= 2) { + SECTION("Changing dst allocation device") { HIP_CHECK(hipSetDevice(1)); LinearAllocGuard new_var(LinearAllocs::hipMalloc, sizeof(int)); HIP_CHECK_ERROR(hipGraphExecMemcpyNodeSetParamsFromSymbol( - graph_exec, node, new_var.ptr(), SYMBOL(int_device_var), - sizeof(*new_var.ptr()), 0, static_cast(-1)), + graph_exec, node, new_var.ptr(), SYMBOL(int_device_var), + sizeof(*new_var.ptr()), 0, static_cast(-1)), hipErrorInvalidValue); + } } HIP_CHECK(hipGraphExecDestroy(graph_exec));