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

[general][filesystem] added structure and namespace to code generation files

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-07 21:15:54 -07:00
parent 392b55280d
commit 7578c27d3d
35 changed files with 224 additions and 147 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -130,7 +130,7 @@ public:
// create profile
triton::dnn::blocksparse::dot dot(N, params_.K, params_.segments, params_.C, "half", params_.bsize, params_.locks, params_.blocks, OP);
// blocksparse matmul
triton::dnn::base* op = dot.enqueue(stream, {&da, &db, &dc, &dlut}, triton::dnn::NO_TUNING);
triton::dnn::base* op = dot.enqueue(stream, {&da, &db, &dc, &dlut}, triton::dnn::PARTIAL_TUNING);
triton::driver::buffer* locks_buffer = ((triton::dnn::blocksparse::dot*)op)->get_locks();
Tensor *tmp = nullptr;
TensorShape tmp_shapes;