From a5235206e74f2fbab23fb6708885c25a7f4e51cd Mon Sep 17 00:00:00 2001
From: Ruobing Han <hanruobing@gatech.edu>
Date: Sun, 25 Sep 2022 12:58:58 -0400
Subject: [PATCH] update CTest

---
 .github/workflows/build.yml | 2 +-
 CMakeLists.txt              | 2 ++
 tests/CMakeLists.txt        | 3 ++-
 tests/runHeteroMark.sh      | 6 ++++--
 4 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 67a8a95..81b0f27 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -35,7 +35,7 @@ jobs:
         run: |
           mkdir build
           cd build
-          cmake .. -DLLVM_CONFIG_PATH=`which llvm-config` -DCUDA_PATH=${{ github.workspace }}/cuda-10.1
+          cmake .. -DLLVM_CONFIG_PATH=`which llvm-config` -DCUDA_PATH=${{ github.workspace }}/cuda-10.1 -DHETERO_MARK_DATA=${{ github.workspace }}/data
           make -j8
       - name: clone SC evaluation code
         run: |
diff --git a/CMakeLists.txt b/CMakeLists.txt
index a5bc849..9f22faa 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -53,4 +53,6 @@ set(GCC_COVERAGE_LINK_FLAGS
 add_subdirectory(compilation)
 add_subdirectory(runtime)
 enable_testing()
+
+option(HETERO_MARK_DATA "The path to download hetero-mark dataset." /tmp/data)
 add_subdirectory(tests)
diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 2349e95..cf2fe9a 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -18,5 +18,6 @@ foreach(case IN LISTS TESTCASES)
     NAME "${case}"
     COMMAND
       ${BASH_PROGRAM} ${CMAKE_CURRENT_SOURCE_DIR}/runHeteroMark.sh ${case}
-      ${CUDA_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/Hetero-Mark ${CMAKE_BINARY_DIR})
+      ${CUDA_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/Hetero-Mark ${CMAKE_BINARY_DIR}
+      ${HETERO_MARK_DATA})
 endforeach()
diff --git a/tests/runHeteroMark.sh b/tests/runHeteroMark.sh
index 84e9992..416e1ed 100644
--- a/tests/runHeteroMark.sh
+++ b/tests/runHeteroMark.sh
@@ -4,12 +4,14 @@ TestCase=$1
 CUDA_PATH=$2
 HeteroMark_PATH=$3
 CuPBoP_BUILD_PATH=$4
-DATASET_PATH=/tmp/data
+DATASET_PATH=$5
 if [ ! -d $DATASET_PATH ]; then
     echo "Download Hetero Mark dataset"
-    cd /tmp
+    mkdir $DATASET_PATH
+    cd $DATASET_PATH
     wget https://www.dropbox.com/s/ie2hcxw9lfoghg8/data.tar.gz?dl=1
     tar -xzf 'data.tar.gz?dl=1'
+    mv data/* .
 fi
 
 mkdir -p $TestCase