From cf2c9a2e6d354cc7d02cc7887993a9b9f60334ae Mon Sep 17 00:00:00 2001
From: rkebichi <54912798+rkebichi@users.noreply.github.com>
Date: Wed, 22 Jan 2020 14:12:22 -0500
Subject: [PATCH] Update CMakeLists.txt

---
 src/CMakeLists.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 44b9fd81a3..a3499b0859 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -22,7 +22,7 @@ if ( DEFINED KFD_WRAPPER )
   set ( KFD_LIB_SRC
     ${LIB_DIR}/kfd/kfd_wrapper.cpp
   )
-  execute_process ( COMMAND sh -xc "${ROOT_DIR}/script/gen_ostream_ops.py -in ${HSA_KMT_INC_PATH}/hsakmttypes.h -out ${ROOT_DIR}/inc/kfd_ostream_ops.h" )
+  execute_process ( COMMAND sh -xc "${ROOT_DIR}/script/gen_ostream_ops.py -in ${HSA_KMT_INC_PATH}/hsakmttypes.h -includes hsakmt.h -out ${ROOT_DIR}/inc/kfd_ostream_ops.h" )
   add_library ( ${KFD_LIB} SHARED ${KFD_LIB_SRC} )
   target_include_directories ( ${KFD_LIB} PRIVATE ${LIB_DIR} ${ROOT_DIR} ${ROOT_DIR}/inc ${HSA_RUNTIME_INC_PATH} ${HSA_RUNTIME_HSA_INC_PATH} ${HSA_KMT_INC_PATH} )
   target_link_libraries( ${KFD_LIB} PRIVATE c stdc++ )