[examples] added skeleton for pytorch wrapper

2019-05-03 14:30:06 -04:00
parent 208d1525de
commit 0d694445e6
8 changed files with 181 additions and 15 deletions
--- a/examples/python/pytorch/CMakeLists.txt
+++ b/examples/python/pytorch/CMakeLists.txt
@@ -0,0 +1,6 @@
+find_package(Torch)
+if(${Torch_FOUND})
+    add_library(torch_triton SHARED conv.cpp)
+    target_compile_features(torch_triton PRIVATE cxx_range_for)
+    target_link_libraries(torch_triton "${TORCH_LIBRARIES}")
+endif()