[examples/tensorflow] fixed #include issue

2019-06-05 11:09:41 -07:00
parent 383b5b2a2a
commit 49fcfd6fc7
4 changed files with 3545 additions and 574 deletions
--- a/examples/python/tensorflow/CMakeLists.txt
+++ b/examples/python/tensorflow/CMakeLists.txt
@@ -5,6 +5,6 @@ if(${TensorFlow_FOUND})
  include_directories("${CUDA_HOME}/include")
  link_directories(${TF_LIB})
  add_definitions(-D_GLIBCXX_USE_CXX11_ABI=0)
-  add_library(tf_blocksparse SHARED blocksparse.cpp)
+  add_library(tf_blocksparse SHARED dot.cpp)
  target_link_libraries(tf_blocksparse tensorflow_framework triton)
 endif()