squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix typo in reference conv3d

Browse Source
This commit is contained in:
Bernardo Covas 2021-05-28 21:06:59 +01:00
parent 9cb7d63424
commit 1d8372a8e2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/util/include/cutlass/util/reference/device/convolution.h
View File

@ -273,7 +273,7 @@ __global__ void Conv3dFprop(
int filter_s = S;
if (problem_size.mode == cutlass::conv::Mode::kConvolution) {
filter_t = problem_size.T - 1 - R;
filter_t = problem_size.T - 1 - T;
filter_r = problem_size.R - 1 - R;
filter_s = problem_size.S - 1 - S;
}