1 files changed, 42 insertions, 0 deletions
diff --git a/infra/nnfw/cmake/packages/TRIXEngineConfig.cmake b/infra/nnfw/cmake/packages/TRIXEngineConfig.cmake
new file mode 100644
index 000000000..dfc10ebf2
--- /dev/null
+++ b/infra/nnfw/cmake/packages/TRIXEngineConfig.cmake
@@ -0,0 +1,42 @@
+# Looking for pre-installed TRIX engine package
+set(TRIX_ENGINE_PREFIX "/usr" CACHE PATH "Where to find TRIX engine header and library")
+
+function(_TRIXEngine_import)
+  # Find the header & lib
+  find_library(TRIXEngine_LIB
+    NAMES npu-engine
+    PATHS "${TRIX_ENGINE_PREFIX}/lib"
+  )
+
+  find_path(TRIXEngine_INCLUDE_DIR
+    NAMES libnpuhost.h
+    PATHS "${TRIX_ENGINE_PREFIX}/include/npu-engine"
+  )
+
+  set(TRIXEngine_FOUND TRUE)
+
+  if(NOT TRIXEngine_LIB)
+    set(TRIXEngine_FOUND FALSE)
+  endif(NOT TRIXEngine_LIB)
+
+  if(NOT TRIXEngine_INCLUDE_DIR)
+    set(TRIXEngine_FOUND FALSE)
+  endif(NOT TRIXEngine_INCLUDE_DIR)
+
+  if(NOT TRIXEngine_FOUND)
+    message(STATUS "Failed to find TRIX Engine")
+  else(NOT TRIXEngine_FOUND)
+
+    # Add target
+    if(NOT TARGET trix_engine)
+      add_library(trix_engine INTERFACE)
+      target_link_libraries(trix_engine INTERFACE ${TRIXEngine_LIB})
+      target_include_directories(trix_engine INTERFACE ${TRIXEngine_INCLUDE_DIR})
+    endif(NOT TARGET trix_engine)
+  endif(NOT TRIXEngine_FOUND)
+
+  set(TRIXEngine_FOUND ${TRIXEngine_FOUND} PARENT_SCOPE)
+  set(TRIXEngine_INCLUDE_DIRS ${TRIXEngine_INCLUDE_DIR} PARENT_SCOPE)
+endfunction(_TRIXEngine_import)
+
+_TRIXEngine_import()