diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-04-24 15:14:12 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:54 +0000 |
| commit | 82b51482479951cf133c223eb81aae291cb4d590 (patch) | |
| tree | f7ce7b0f169755234f788d68e296a769230e9964 /src/graph | |
| parent | 35aa6a36e7ef9302efd554aac9b84153ad7c6a9e (diff) | |
| download | ComputeLibrary-82b51482479951cf133c223eb81aae291cb4d590.tar.gz | |
COMPMID-959: Sets memory manager to CLWinograd
-Sets memory manager to Winograd functions
-Marks CLGEMM inputs as unused if needed
Change-Id: I425a3f864c756e0e2b4da895e1730b8822149ba8
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128891
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/graph')
| -rw-r--r-- | src/graph/backends/CL/CLFunctionsFactory.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/graph/backends/CL/CLFunctionsFactory.cpp b/src/graph/backends/CL/CLFunctionsFactory.cpp index ad73a797e3..ece63646ea 100644 --- a/src/graph/backends/CL/CLFunctionsFactory.cpp +++ b/src/graph/backends/CL/CLFunctionsFactory.cpp @@ -174,8 +174,8 @@ std::unique_ptr<IFunction> create_convolution_layer(ConvolutionLayerNode &node, if(conv_algorithm == ConvolutionMethod::WINOGRAD) { - std::tie(func, func_name) = create_named_function<CLWinogradConvolutionLayer>( - std::string("CLWinogradConvolutionLayer"), input, weights, biases, output, conv_info); + std::tie(func, func_name) = create_named_memory_managed_function<CLWinogradConvolutionLayer>( + std::string("CLWinogradConvolutionLayer"), mm, input, weights, biases, output, conv_info); } else if(conv_algorithm == ConvolutionMethod::DIRECT) { |