From 125b738ccb4edc179ec90dc21910eb6a69541393 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Sat, 30 Jan 2021 23:03:49 -0600
Subject: [PATCH] Fixed bugs introduced in 9c11c88033
---
hpvm/test/dnn_benchmarks/hpvm-c/CMakeLists.txt | 5 +++--
hpvm/test/dnn_benchmarks/hpvm-c/include/hpvm.h | 2 +-
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/hpvm/test/dnn_benchmarks/hpvm-c/CMakeLists.txt b/hpvm/test/dnn_benchmarks/hpvm-c/CMakeLists.txt
index 3b78ad26df..961877789a 100644
--- a/hpvm/test/dnn_benchmarks/hpvm-c/CMakeLists.txt
+++ b/hpvm/test/dnn_benchmarks/hpvm-c/CMakeLists.txt
@@ -20,12 +20,13 @@ set(CONFIG_INCLUDE_DIR "${CMAKE_CURRENT_BINARY_DIR}/include")
# Directories to include
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_RT_INCLUDE_DIR ${HPVM_PROJECTS}/hpvm-tensor-rt/tensor_runtime/include)
set(
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}
)
foreach(dir ${INCLUDES})
diff --git a/hpvm/test/dnn_benchmarks/hpvm-c/include/hpvm.h b/hpvm/test/dnn_benchmarks/hpvm-c/include/hpvm.h
index 07ccd05b55..0587311910 100644
--- a/hpvm/test/dnn_benchmarks/hpvm-c/include/hpvm.h
+++ b/hpvm/test/dnn_benchmarks/hpvm-c/include/hpvm.h
@@ -10,7 +10,7 @@
#define DEVICE CUDNN_TARGET
#endif
-#include "../../../include/SupportHPVM/HPVMHint.h"
+#include "SupportHPVM/HPVMHint.h"
#ifdef __cplusplus
extern "C" {
--
GitLab