gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[examples/tensorflow] fixed #include issue

Browse Source
This commit is contained in:
Philippe Tillet
2019-06-05 11:09:41 -07:00
parent 383b5b2a2a
commit 49fcfd6fc7
4 changed files with 3545 additions and 574 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/python/tensorflow/CMakeLists.txt
View File

@@ -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()

2
examples/python/tensorflow/dot.cpp
View File

@@ -3,7 +3,7 @@
#include "triton/driver/buffer.h"
#include "triton/driver/backend.h"
#include "triton/driver/stream.h"
#include "triton/jit.h"
#include "triton/runtime/jit.h"
#define EIGEN_USE_GPU
#include "tensorflow/core/framework/op.h"