include_directories(
  ${TORCH_INCLUDE_DIRS}
  )
add_library(aten_ops SHARED
  aten_ops.cpp
  )

target_link_libraries(aten_ops
  ${TORCH_LIBRARIES}
 )