"profiler/vscode:/vscode.git/clone" did not exist on "8f8c0ddc58b190f4b1cfa35623734493fadc5fa9"
Commit d7a829f4 authored by Paul's avatar Paul
Browse files

Select correct workspace memory for convolution

parent 0418f719
......@@ -129,8 +129,7 @@ struct miopen_convolution
workspace_size,
false);
algo = perf.fwd_algo;
return algo == miopenConvolutionFwdAlgoWinograd ? shape{shape::int8_type, {0}}
: workspace_shape;
return shape{shape::int8_type, {perf.memory}};
}
};
......
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