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

stuff

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-03 19:25:16 -07:00
parent 0d8faa5b1e
commit 1d88f0a36b
7 changed files with 194 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/cpp/shift.cpp
View File

@@ -74,12 +74,12 @@ int main() {
// shift
std::vector<unsigned> params = {
16, 4, 64, 16, 4, 128, 2, 2, 1, 2, 4, 4, 16, 4
4, 2, 16, 4, 128, 2, 2, 1, 1, 8, 16, 8, 2
};
std::ostringstream oss;
shift.src(oss);
std::string src = oss.str();
jit.autotune("shift", src.c_str(), benchmark);
// jit.autotune("shift", src.c_str(), benchmark);
jit.add_module("shift", src.c_str(), params);
triton::driver::kernel* kernel = jit.get_function("shift");
triton::jit::launch_information info = jit.get_launch_info("shift");