Skip col_buffer allocation when not needed#1171
Conversation
|
Thanks for PRing this. However, see my comment #1118 (comment); I believe your expression is wrong and breaks convolution in certain cases. Also, I think we should test this rather more aggressively in terms of the parameter space than the existing conv tests (even if we only do this once, offline) as in any case this will add significant subtlety to the code paths. |
|
@longjon agreed, we need a more complete set of tests. Currently I only need the first two cases 1x1 and Single, for which I have added test cases, so we can keep that two in this PR and leave the other 2 for later, or I add them now and test them offline (I think they a less frequent use case). |
dc1ac9d to
8ff90d0
Compare
8ff90d0 to
acd0276
Compare
|
(See #1118 (comment), and let's continue the discussion here.) If it's not too onerous to gain confidence in its correctness, we might as well use the general expression now. Since all the tests are for |
|
@longjon could we merge this? |
|
@longjon although we could compute |
|
Closing since the dev branch is deprecated. Please send PRs to master. |
In the line of #1118 this PR based on the discussion with @longjon and @shelhamer extends the cases when the col_buffer is not needed and a matrix multiplication is enough to compute the result.