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/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){
std::string B = std::to_string(B_);
std::string CW = std::to_string(CW_);
std::string CH = std::to_string(CH_);
std::string CW = std::to_string(ICW_);
std::string CH = std::to_string(ICH_);
if(is_chwn) {
return R"(