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

Merge pull request #272 from BernardoCovas/master

Browse Source 
Bug in reference conv3d
This commit is contained in:
Haicheng Wu 2021-05-28 17:18:27 -04:00 committed by GitHub
commit ad9486684f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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;
}