Skip to content
Snippets Groups Projects
Commit 125b738c authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Fixed bugs introduced in 9c11c880

parent 529e3cd9
No related branches found
No related tags found
No related merge requests found
...@@ -20,12 +20,13 @@ set(CONFIG_INCLUDE_DIR "${CMAKE_CURRENT_BINARY_DIR}/include") ...@@ -20,12 +20,13 @@ set(CONFIG_INCLUDE_DIR "${CMAKE_CURRENT_BINARY_DIR}/include")
# Directories to include # Directories to include
set(HPVM_PROJECTS ${PROJECT_SOURCE_DIR}/tools/hpvm/projects) set(HPVM_PROJECTS ${PROJECT_SOURCE_DIR}/tools/hpvm/projects)
set(HPVM_INCLUDE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../dnn_benchmarks/include) set(HPVM_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/tools/hpvm/include)
set(BENCHMARK_INCLUDE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/../dnn_benchmarks/include)
set(TENSOR_INCLUDE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/include) set(TENSOR_INCLUDE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/include)
set(TENSOR_RT_INCLUDE_DIR ${HPVM_PROJECTS}/hpvm-tensor-rt/tensor_runtime/include) set(TENSOR_RT_INCLUDE_DIR ${HPVM_PROJECTS}/hpvm-tensor-rt/tensor_runtime/include)
set( set(
INCLUDES INCLUDES
${CONFIG_INCLUDE_DIR} ${HPVM_INCLUDE_DIR} ${CONFIG_INCLUDE_DIR} ${HPVM_INCLUDE_DIR} ${BENCHMARK_INCLUDE_DIR}
${TENSOR_INCLUDE_DIR} ${TENSOR_RT_INCLUDE_DIR} ${CUDA_INCLUDE_DIRS} ${TENSOR_INCLUDE_DIR} ${TENSOR_RT_INCLUDE_DIR} ${CUDA_INCLUDE_DIRS}
) )
foreach(dir ${INCLUDES}) foreach(dir ${INCLUDES})
......
...@@ -10,7 +10,7 @@ ...@@ -10,7 +10,7 @@
#define DEVICE CUDNN_TARGET #define DEVICE CUDNN_TARGET
#endif #endif
#include "../../../include/SupportHPVM/HPVMHint.h" #include "SupportHPVM/HPVMHint.h"
#ifdef __cplusplus #ifdef __cplusplus
extern "C" { extern "C" {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment