Commit e2630858 authored by Paul's avatar Paul
Browse files

Merge branch 'workspace'

parents f8edb479 d7a829f4
...@@ -129,8 +129,7 @@ struct miopen_convolution ...@@ -129,8 +129,7 @@ struct miopen_convolution
workspace_size, workspace_size,
false); false);
algo = perf.fwd_algo; algo = perf.fwd_algo;
return algo == miopenConvolutionFwdAlgoWinograd ? shape{shape::int8_type, {0}} return shape{shape::int8_type, {perf.memory}};
: workspace_shape;
} }
}; };
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment