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/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"