From d2ea4d3dd44f59fb04c41b6e76a0169b4d5c2685 Mon Sep 17 00:00:00 2001
From: sdashmiz <shadi.dashmiz@amd.com>
Date: Mon, 6 Mar 2023 11:39:14 -0500
Subject: [PATCH] SWDEV-385255 - fixes for graph exec with mem pools

Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Change-Id: I6f127090904a6c7985992bfaaab928d8b03013f5
---
 hipamd/src/hip_graph_internal.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hipamd/src/hip_graph_internal.cpp b/hipamd/src/hip_graph_internal.cpp
index 3c118bc198..847c72f9d0 100644
--- a/hipamd/src/hip_graph_internal.cpp
+++ b/hipamd/src/hip_graph_internal.cpp
@@ -574,7 +574,7 @@ hipError_t hipGraphExec::Run(hipStream_t stream) {
   if (hip::getStream(stream) == nullptr) {
     return hipErrorInvalidResourceHandle;
   }
-  if (flags_ == hipGraphInstantiateFlagAutoFreeOnLaunch) {
+  if (flags_ & hipGraphInstantiateFlagAutoFreeOnLaunch) {
     if (!levelOrder_.empty()) {
       levelOrder_[0]->GetParentGraph()->FreeAllMemory();
     }