From 1914db71de52a9a8b608d83814098657e86b3bcc Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Fri, 29 Sep 2017 12:05:49 +0100 Subject: COMPMID-417 - Fix graph_letnet.cpp example In case the library has been built just for NEON, we should not run the example Change-Id: I99f8d3efd20276dfeb73ca8362872111be1b1fd7 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/89682 Reviewed-by: Anthony Barbier Tested-by: Kaizen --- examples/graph_lenet.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'examples/graph_lenet.cpp') diff --git a/examples/graph_lenet.cpp b/examples/graph_lenet.cpp index 676fdb9ce2..1f04ff8a50 100644 --- a/examples/graph_lenet.cpp +++ b/examples/graph_lenet.cpp @@ -93,16 +93,18 @@ void main_graph_lenet(int argc, const char **argv) } // Check if OpenCL is available and initialize the scheduler + Hint hint = Hint::NEON; if(arm_compute::opencl_is_available()) { arm_compute::CLScheduler::get().default_init(); + hint = Hint::OPENCL; } Graph graph; graph.set_info_enablement(true); //conv1 << pool1 << conv2 << pool2 << fc1 << act1 << fc2 << smx - graph << Hint::OPENCL + graph << hint << Tensor(TensorInfo(TensorShape(28U, 28U, 1U, batches), 1, DataType::F32), DummyAccessor()) << ConvolutionLayer( 5U, 5U, 20U, -- cgit v1.2.1