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

[triton/python/conv]: Added cache for compiled kernels

Browse Source
This commit is contained in:
Philippe Tillet
2019-05-18 11:51:49 -04:00
parent 600aef72d5
commit b2b55c52c9
10 changed files with 210 additions and 516 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/cpp/shift.cpp
View File

@@ -1,7 +1,7 @@
#include <cstring>
#include <cstdio>
#include "common.hpp"
#include "triton/jit.h"
#include "triton/runtime/jit.h"
#include "triton/driver/backend.h"
#include "triton/driver/stream.h"
@@ -158,8 +158,8 @@ int main() {
unsigned TN = info.global_range_size[1];
unsigned nthreads = info.num_threads;
// initialize constant memory
triton::driver::buffer* delta = jit.get_buffer("delta");
triton::driver::buffer* masks = jit.get_buffer("masks");
triton::driver::buffer* delta = ((triton::driver::cu_module*)kernel->module())->symbol("delta");
triton::driver::buffer* masks = ((triton::driver::cu_module*)kernel->module())->symbol("masks");
stream->write(delta, false, 0, h_delta.size()*4, h_delta.data());
stream->write(masks, false, 0, h_masks.size()*4, h_masks.data());
stream->synchronize();