From e7348b1f041f5c59268c99c263de9cab99d6708c Mon Sep 17 00:00:00 2001
From: ROCm CI Service Account <66695075+rocm-ci@users.noreply.github.com>
Date: Wed, 21 Jun 2023 15:58:27 +0530
Subject: [PATCH] SWDEV-388950 - updating test (#250)

Change-Id: I402542b53a84203a8f047534e40e3571bf1eb0c3
---
 .../hipGraphExecMemcpyNodeSetParamsFromSymbol.cc     | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc b/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc
index 03fa33636c..ea65bd4ead 100644
--- a/catch/unit/graph/hipGraphExecMemcpyNodeSetParamsFromSymbol.cc
+++ b/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<int> 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<hipMemcpyKind>(-1)),
+                    graph_exec, node, new_var.ptr(), SYMBOL(int_device_var),
+                    sizeof(*new_var.ptr()), 0, static_cast<hipMemcpyKind>(-1)),
                     hipErrorInvalidValue);
+    }
   }
 
   HIP_CHECK(hipGraphExecDestroy(graph_exec));