From 7215451db9372c9c61cf2edfe64a21c002aa4190 Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Mon, 10 Apr 2023 07:24:18 +0000
Subject: [PATCH] SWDEV-388958 - SWDEV-388929 - Kind can be hipMemcpyDefault
 for H2D and D2D and srcMemory can be allocated using hipHostMalloc for H2D.

Change-Id: I187ff850ebb890b357ee03500105b71468508ea2
---
 hipamd/src/hip_graph_internal.hpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/hipamd/src/hip_graph_internal.hpp b/hipamd/src/hip_graph_internal.hpp
index 3aab90081c..479819620b 100644
--- a/hipamd/src/hip_graph_internal.hpp
+++ b/hipamd/src/hip_graph_internal.hpp
@@ -1533,9 +1533,10 @@ class hipGraphMemcpyNodeToSymbol : public hipGraphMemcpyNode1D {
     }
     size_t dOffset = 0;
     amd::Memory* srcMemory = getMemoryObject(src, dOffset);
-    if (srcMemory == nullptr && kind != hipMemcpyHostToDevice) {
+    if (srcMemory == nullptr && kind != hipMemcpyHostToDevice && kind != hipMemcpyDefault) {
       return hipErrorInvalidValue;
-    } else if (srcMemory != nullptr && kind != hipMemcpyDeviceToDevice) {
+    } else if (srcMemory != nullptr && srcMemory->getMemFlags() == 0 &&
+               kind != hipMemcpyDeviceToDevice && kind != hipMemcpyDefault) {
       return hipErrorInvalidValue;
     } else if (kind == hipMemcpyHostToHost || kind == hipMemcpyDeviceToHost) {
       return hipErrorInvalidValue;