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

seleciton now segfault (expected

Browse Source
This commit is contained in:
Philippe Tillet
2019-06-06 20:34:56 -07:00
parent 6045209d5b
commit 781b6d377d
2 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -155,7 +155,10 @@ class BlockSparseGemmOp : public OpKernel {
return 2.*M*N*K / ts * 1e-3;
};
// just-in-time compile source-code
jit.autotune("matmul", src, benchmark);
jit.add_module("matmul", src, {4, 2, 16, 4, 2, 16, 2, 2, 1, 1, 8, 8, 8, 1});
triton::driver::kernel* kernel = jit.get_function("matmul");
triton::jit::launch_information info = jit.get_launch_info("matmul");
benchmark(kernel, info);
}
private:

8
lib/runtime/jit.cpp
View File

@@ -121,14 +121,16 @@ jit::tune_res_t jit::autotune(const char *name, const char *src, benchmark_t ben
passes.target_independent(tt_module);
passes.tune.init(tt_module);
passes.tune.check_constraints(errors);
for(auto x: errors)
for(auto err: x.second)
std::cout << err << std::endl;
// for(auto x: errors)
// for(auto err: x.second)
// std::cout << err << std::endl;
if(!errors.empty())
return;
// Deep copy of the module and tuner
auto ptt_module = make_triton_module(name, src);
ir::module &tt_module = *ptt_module;
for(unsigned p: params)
std::cout << p << " " << std::flush;
passes_wrapper passes(target_.get());
passes.target_independent(tt_module);
passes.tune.run(tt_module);