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

[dnn/shift] bugfix in static shape division

Browse Source
This commit is contained in:
Philippe Tillet
2019-07-17 11:39:17 -07:00
parent a55b098e88
commit 791c91ee63
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/codegen/tune.cpp
View File

@@ -223,7 +223,7 @@ void tune::run(ir::module &mod) {
} }
else { else {
ir::metaparameter *fpw = ir::metaparameter::create(ctx, ty, 2, 2); ir::metaparameter *fpw = ir::metaparameter::create(ctx, ty, 2, 2);
ir::metaparameter *wpt = ir::metaparameter::create(ctx, ty, 2, 4); ir::metaparameter *wpt = ir::metaparameter::create(ctx, ty, 1, 4);
connected_components(node, {fpw, wpt}, {"fpw", "wpt"}, nodes_, dependencies_, group_id++); connected_components(node, {fpw, wpt}, {"fpw", "wpt"}, nodes_, dependencies_, group_id++);
} }
} }
@@ -237,7 +237,7 @@ void tune::run(ir::module &mod) {
continue; continue;
if(dynamic_cast<ir::load_inst*>(i) && i->get_type()->is_tile_ty()){ if(dynamic_cast<ir::load_inst*>(i) && i->get_type()->is_tile_ty()){
ir::type *ty = mod.get_builder().get_int32_ty(); ir::type *ty = mod.get_builder().get_int32_ty();
std::unique_ptr<ir::metaparameter> tmp(ir::metaparameter::create(ctx, ty, 2, 4)); std::unique_ptr<ir::metaparameter> tmp(ir::metaparameter::create(ctx, ty, 2, 2));
*params_.at(i).at("nts.d0") = *tmp; *params_.at(i).at("nts.d0") = *tmp;
} }
if(dynamic_cast<ir::dot_inst*>(i) && i->get_type()->is_tile_ty()){ if(dynamic_cast<ir::dot_inst*>(i) && i->get_type()->is_tile_ty()){

4
lib/dnn/shift.cpp
View File

@@ -292,8 +292,8 @@ void shift::triton_c_src(std::ostream &os) const {
auto compute_bhw = [&](std::string rx, std::string sz, std::string rkx){ auto compute_bhw = [&](std::string rx, std::string sz, std::string rkx){
std::string B = std::to_string(B_); std::string B = std::to_string(B_);
std::string CW = std::to_string(CW_); std::string CW = std::to_string(ICW_);
std::string CH = std::to_string(CH_); std::string CH = std::to_string(ICH_);
if(is_chwn) { if(is_chwn) {
return R"( return R"(