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

[feature] added basic tensor core support

Browse Source
This commit is contained in:
Philippe Tillet
2019-06-09 14:41:36 -07:00
parent d074a166e2
commit 06b5992509
9 changed files with 180 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/runtime/jit.cpp
View File

@@ -129,8 +129,8 @@ jit::tune_res_t jit::autotune(const char *name, const char *src, benchmark_t ben
// 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;
// for(unsigned p: params)
// std::cout << p << " " << std::flush;
passes_wrapper passes(target_.get());
passes.target_independent(tt_module);
passes.tune.run(tt_module);