CMake Patch 1

set(HPVM_TEST_UTILS opt hpvm-rt-bc FileCheck count not)

configure_lit_site_cfg(
  ../lit.site.cfg.py.in
  ${CMAKE_CURRENT_BINARY_DIR}/hpvm_core_pass/lit.site.cfg.py
  MAIN_CONFIG
  ${CMAKE_CURRENT_SOURCE_DIR}/hpvm_core_pass/lit.cfg.py
)

set(TEST_DEPS )
list(APPEND TEST_DEPS ${HPVM_TEST_UTILS} HPVMGenHPVM HPVMBuildDFG HPVMLocalMem HPVMClearDFG
HPVMDFG2LLVM_CPU HPVMDFGTransformPasses HPVMUnrollAndJam HPVMArgPriv
HPVMBufferIn)
if(${HPVM_USE_OPENCL} EQUAL 1)
  list(APPEND TEST_DEPS  HPVMDFG2LLVM_GPU_OCL HPVMDFG2LLVM_FPG)
endif()

add_lit_testsuite(check-hpvm-core-pass "Running test cases for the HPVM core passes"
  ${CMAKE_CURRENT_BINARY_DIR}/hpvm_core_pass
  DEPENDS ${TEST_DEPS}
)
add_custom_target(check-hpvm-pass DEPENDS check-hpvm-core-pass)

if(HPVM_USE_CUDA)
  configure_lit_site_cfg(
    ../lit.site.cfg.py.in
    ${CMAKE_CURRENT_BINARY_DIR}/hpvm_tensor_pass/lit.site.cfg.py
    MAIN_CONFIG
    ${CMAKE_CURRENT_SOURCE_DIR}/hpvm_tensor_pass/lit.cfg.py
  )
  add_lit_testsuite(check-hpvm-tensor-pass "Running test cases for the HPVM tensor passes"
    ${CMAKE_CURRENT_BINARY_DIR}/hpvm_tensor_pass
    DEPENDS ${HPVM_TEST_UTILS}
    HPVMDFG2LLVM_CUDNN HPVMDFG2LLVM_WrapperAPI
    HPVMFuseHPVMTensorNodes HPVMInPlaceDFGAnalysis
  )
  add_dependencies(check-hpvm-pass check-hpvm-tensor-pass)
endif()