diff options
Diffstat (limited to 'examples/graph_vgg16.cpp')
| -rw-r--r-- | examples/graph_vgg16.cpp | 24 |
1 files changed, 4 insertions, 20 deletions
diff --git a/examples/graph_vgg16.cpp b/examples/graph_vgg16.cpp index 82be97c316..9e8e69411f 100644 --- a/examples/graph_vgg16.cpp +++ b/examples/graph_vgg16.cpp @@ -32,20 +32,6 @@ using namespace arm_compute::utils; using namespace arm_compute::graph::frontend; using namespace arm_compute::graph_utils; -namespace -{ -/** This function checks if we can use GEMM-based convolution trying to allocate a memory of size "size_in_bytes" - * - * @param[in] size_in_bytes Memory size in bytes needed for VGG-16 - * - * @return The convolution layer hint - */ -ConvolutionMethod convolution_hint_vgg16(size_t size_in_bytes) -{ - return ((get_mem_free_from_meminfo() * 1024) >= size_in_bytes) ? ConvolutionMethod::GEMM : ConvolutionMethod::DIRECT; -} -} // namespace - /** Example demonstrating how to implement VGG16's network using the Compute Library's graph API * * @param[in] argc Number of arguments @@ -65,14 +51,12 @@ public: std::unique_ptr<IPreprocessor> preprocessor = arm_compute::support::cpp14::make_unique<CaffePreproccessor>(mean_rgb); // Set target. 0 (NEON), 1 (OpenCL), 2 (OpenCL with Tuner). By default it is NEON - const int target = argc > 1 ? std::strtol(argv[1], nullptr, 10) : 0; - Target target_hint = set_target_hint(target); + const int target = argc > 1 ? std::strtol(argv[1], nullptr, 10) : 0; + Target target_hint = set_target_hint(target); + const bool is_opencl = target_hint == Target::CL; - // Check if we can use GEMM-based convolutions evaluating if the platform has at least 1.8 GB of available memory - const size_t memory_required = 1932735283L; - const bool is_opencl = target_hint == Target::CL; ConvolutionMethod first_convolution3x3_hint = is_opencl ? ConvolutionMethod::DIRECT : ConvolutionMethod::GEMM; - ConvolutionMethod convolution3x3_hint = is_opencl ? ConvolutionMethod::WINOGRAD : convolution_hint_vgg16(memory_required); + ConvolutionMethod convolution3x3_hint = ConvolutionMethod::DEFAULT; // Parse arguments if(argc < 2) |