[BYOC][CONTRIB] Vitis-AI codegen integration

CMakeLists.txt

@@ -75,7 +75,7 @@ tvm_option(USE_COREML "Build with coreml support" OFF)
 tvm_option(USE_TARGET_ONNX "Build with ONNX Codegen support" OFF)
 tvm_option(USE_ARM_COMPUTE_LIB "Build with Arm Compute Library" OFF)
 tvm_option(USE_ARM_COMPUTE_LIB_GRAPH_RUNTIME "Build with Arm Compute Library graph runtime" OFF)
-
+tvm_option(USE_VITIS_AI "Build with VITIS-AI Codegen support" OFF)   
 
 # include directories
 include_directories(${CMAKE_INCLUDE_PATH})
@@ -323,6 +323,7 @@ include(cmake/modules/contrib/TF_TVMDSOOP.cmake)
 include(cmake/modules/contrib/CoreML.cmake)
 include(cmake/modules/contrib/ONNX.cmake)
 include(cmake/modules/contrib/ArmComputeLib.cmake)
+include(cmake/modules/contrib/VITISAI.cmake) 
 include(cmake/modules/Git.cmake)
 include(cmake/modules/LibInfo.cmake)
 

cmake/config.cmake

@@ -220,6 +220,9 @@ set(USE_ETHOSN OFF)
 # otherwise use ETHOSN_HW (OFF) to use the software test infrastructure
 set(USE_ETHOSN_HW OFF)
 
+# Whether use VITIS-AI codegen
+set(USE_VITIS_AI OFF)
+
 # Build ANTLR parser for Relay text format
 # Possible values:
 # - ON: enable ANTLR by searching default locations (cmake find_program for antlr4 and /usr/local for jar)

cmake/modules/contrib/VITISAI.cmake

@@ -0,0 +1,49 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+if(USE_VITIS_AI)
+  set(PYXIR_SHARED_LIB libpyxir.so)
+  find_package(PythonInterp 3.6 REQUIRED)
+  if(NOT PYTHON)
+    find_program(PYTHON NAMES python3 python3.6)
+  endif()
+  if(PYTHON)
+    execute_process(COMMAND "${PYTHON_EXECUTABLE}" "-c"
+      "import pyxir as px; print(px.get_include_dir()); print(px.get_lib_dir());"
+      RESULT_VARIABLE __result
+      OUTPUT_VARIABLE __output
+      OUTPUT_STRIP_TRAILING_WHITESPACE)
+
+    if(__result MATCHES 0)
+      string(REGEX REPLACE ";" "\\\\;" __values ${__output})
+      string(REGEX REPLACE "\r?\n" ";"    __values ${__values})
+      list(GET __values 0 PYXIR_INCLUDE_DIR)
+      list(GET __values 1 PYXIR_LIB_DIR)
+    endif()
+
+  else()
+  message(STATUS "To find Pyxir, Python interpreter is required to be found.")
+endif()
+
+message(STATUS "Build with contrib.vitisai")
+include_directories(${PYXIR_INCLUDE_DIR})  
+file(GLOB VAI_CONTRIB_SRC src/runtime/contrib/vitis_ai/*.cc)
+link_directories(${PYXIR_LIB_DIR})
+list(APPEND TVM_RUNTIME_LINKER_LIBS "pyxir")
+list(APPEND RUNTIME_SRCS ${VAI_CONTRIB_SRC})
+endif(USE_VITIS_AI)
+