FILE(GLOB SOURCES src/*.cpp)

add_library(torch_modules STATIC ${SOURCES})
target_link_libraries(torch_modules "${TORCH_LIBRARIES}")