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

Reparameterized in terms of micro- and nano- tiles

Browse Source
This commit is contained in:
Philippe Tillet
2019-03-10 23:10:17 -04:00
parent c96a263896
commit 94e315ea8a
7 changed files with 62 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/jit.cpp
View File

@@ -86,7 +86,6 @@ std::unique_ptr<llvm::Module> jit::make_llvm_module(ir::module &module, codegen:
// constraints
std::map<ir::value*, std::vector<std::string>> errors;
tune.check_constraints(module, errors);
std::cout << "errors: " << errors.size() << std::endl;
for(auto &x: errors){
for(auto &e: x.second)
std::cout << x.first->get_name() << " " << e << std::endl;
@@ -150,7 +149,13 @@ void jit::autotune(ir::module &tt_module, benchmark_t benchmark) {
tune.check_constraints(tt_module, errors);
if(errors.size())
return;
std::cout << "valid" << std::endl;
ir::module copy(tt_module);
auto ll_module = make_llvm_module(copy, tune);
driver::module module(driver_context_, &*ll_module);
driver::kernel kernel(module, "matmul");
launch_information info = launch_info_map_.at("matmul");
benchmark(kernel, info);
std::cout << "benchmarked" << std::endl;
});
}