CMAKE_MINIMUM_REQUIRED(VERSION 3.8.2 FATAL_ERROR)
PROJECT(torch_cg)

SET(CMAKE_CXX_STANDARD 14)
SET(CMAKE_CXX_STANDARD_REQUIRED ON)
SET(CMAKE_POSITION_INDEPENDENT_CODE ON)

ADD_SUBDIRECTORY(pybind11)

SET(MYLIB "/content/torch_cg/sample")
#SET(MYLIB "/Users/Jeff/SFU/PHD/DATASET_BUILDER_V2/torch_cg/sample")
INCLUDE_DIRECTORIES( "${MYLIB}" )

FIND_PACKAGE(Protobuf REQUIRED)
INCLUDE_DIRECTORIES(${Protobuf_INCLUDE_DIR})

PYBIND11_ADD_MODULE(torch_cg main.cpp midi.pb.cc)

TARGET_LINK_LIBRARIES(torch_cg PRIVATE "${Protobuf_LIBRARY}" "${MYLIB}/build/libsample.so")