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

[code generation] uniformized shape and layout metaparameters

Browse Source
This commit is contained in:
Philippe Tillet
2019-03-09 12:31:21 -05:00
parent 5f29263044
commit b721202812
3 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/jit.cpp
View File

@@ -62,12 +62,9 @@ std::unique_ptr<llvm::Module> jit::make_llvm_module(ir::module &module, const st
// tuning parameters
tune.run(module);
unsigned i = 0;
triton_context_.p_impl->mp_constants_[0]->set_value(params[0]);
triton_context_.p_impl->mp_constants_[1]->set_value(params[1]);
triton_context_.p_impl->mp_constants_[2]->set_value(params[2]);
for(ir::metaparameter *x: tune.get_params(module)){
x->set_value(params[3 + i++]);
}
for(ir::metaparameter *x: tune.get_params(module))
x->set_value(params[i++]);
// constraints
std::map<ir::value*, std::vector<std::string>> errors;
tune.check_constraints(module, errors);