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

[jit] changed default metaparameter ranges

Browse Source
This commit is contained in:
Philippe Tillet
2019-03-10 10:45:21 -04:00
parent d2e7d7890d
commit c96a263896
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/matrix.cpp
View File

@@ -123,6 +123,7 @@ int main() {
};
triton::jit jit(context);
jit.add_module(src, params);
jit.autotune(src, benchmark);
triton::driver::kernel kernel = jit.get_function("matmul");
triton::jit::launch_information info = jit.get_launch_info("matmul");
benchmark(kernel, info);

2
lib/ast/lowering.cpp
View File

@@ -412,7 +412,7 @@ ir::value* initializer::codegen(ir::module * mod) const{
if(std::find(storage.begin(), storage.end(), TUNABLE_T) != storage.end()){
assert(expr_ == nullptr);
//TODO: implement ranges
value = ir::metaparameter::create(mod->get_context(), ty, 4, 8);
value = ir::metaparameter::create(mod->get_context(), ty, 8, 64);
}
if(expr_){
value = expr_->codegen(mod);

6
lib/codegen/tune.cpp
View File

@@ -145,9 +145,9 @@ void tune::run(ir::module &mod) {
// Layout parameters
while(!nodes_.empty()){
ir::type *ty = mod.get_builder().get_int32_ty();
ir::metaparameter *mp0 = ir::metaparameter::create(ctx, ty, 1, 4);
ir::metaparameter *mp1 = ir::metaparameter::create(ctx, ty, 4, 32);
ir::metaparameter *mp2 = ir::metaparameter::create(ctx, ty, 4, 32);
ir::metaparameter *mp0 = ir::metaparameter::create(ctx, ty, 2, 2);
ir::metaparameter *mp1 = ir::metaparameter::create(ctx, ty, 4, 8);
ir::metaparameter *mp2 = ir::metaparameter::create(ctx, ty, 1, 4);
connected_components(*nodes_.begin(), {mp0, mp1, mp2}, nodes_, dependencies_);
}
}