From eb82fd2aa786715c3b6a941dc6d6deac4ce8e2a0 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Fri, 23 Feb 2018 13:43:50 +0000 Subject: COMPMID-881: RSH new arm_gemm interface. Change-Id: I1e2a1a77097d8017c274af3f97eba6964f80f5fa Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122592 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- examples/graph_inception_v4.cpp | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'examples') diff --git a/examples/graph_inception_v4.cpp b/examples/graph_inception_v4.cpp index d9f6156fb2..6f76b5e2e7 100644 --- a/examples/graph_inception_v4.cpp +++ b/examples/graph_inception_v4.cpp @@ -21,6 +21,7 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ +#include "arm_compute/core/Error.h" #include "arm_compute/graph2.h" #include "support/ToolchainSupport.h" #include "utils/GraphUtils.h" @@ -43,6 +44,9 @@ class InceptionV4Example final : public Example public: void do_setup(int argc, char **argv) override { + // Disabled the test for now because the process gets killed on Linux Firefly 32 bit even when using ConvolutionMethodHint::DIRECT. + // Needs to review/rework to run the code below. +#if __aarch64__ std::string data_path; /* Path to the trainable data */ std::string image; /* Image data */ std::string label; /* Label data */ @@ -90,7 +94,6 @@ public: graph << target_hint << InputLayer(TensorDescriptor(TensorShape(299U, 299U, 3U, 1U), DataType::F32), get_input_accessor(image, std::move(preprocessor), false)) - // Conv2d_1a_3x3 << ConvolutionLayer(3U, 3U, 32U, get_weights_accessor(data_path, "/cnn_data/inceptionv4_model/Conv2d_1a_3x3_weights.npy"), @@ -157,11 +160,18 @@ public: // Finalize graph graph.finalize(target_hint, enable_tuning, enable_memory_management); +#else /* __aarch64__ */ + using namespace arm_compute; + ARM_COMPUTE_UNUSED(argc); + ARM_COMPUTE_UNUSED(argv); +#endif /* __aarch64__ */ } void do_run() override { +#if __aarch64__ graph.run(); +#endif /* __aarch64__ */ } private: -- cgit v1.2.1