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

[dnn/conv] Added the option to have look-up table for filters for all

Browse Source 
operations
This commit is contained in:
Philippe Tillet
2019-05-22 19:03:33 -04:00
parent f8291af7ef
commit 3f3eb1c2a4
3 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/dnn/conv.cpp
View File

@@ -301,12 +301,12 @@ void conv::set_arg(driver::kernel *kernel,
}
std::vector<unsigned> conv::default_params() {
if(ty_==FPROP)
if(b_lut_)
return {32, 2, 64, 32, 2, 64, 16, 8, 2, 2, 4, 2, 8};
else if(ty_ == FPROP)
return {16, 2, 64, 32, 2, 64, 16, 8, 2, 2, 8, 1, 8, 4};
else if(ty_ == BPROP)
return {32, 2, 64, 32, 64, 32, 4, 2, 2, 4, 2, 8, 4, 2};
else if(ty_ == WGRAD)
return {32, 2, 64, 32, 2, 64, 16, 8, 2, 2, 4, 2, 8};
}