196775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet/* 296775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * Copyright (C) 2017 The Android Open Source Project 396775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * 496775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * Licensed under the Apache License, Version 2.0 (the "License"); 596775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * you may not use this file except in compliance with the License. 696775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * You may obtain a copy of the License at 796775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * 896775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * http://www.apache.org/licenses/LICENSE-2.0 996775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * 1096775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * Unless required by applicable law or agreed to in writing, software 1196775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * distributed under the License is distributed on an "AS IS" BASIS, 1296775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1396775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * See the License for the specific language governing permissions and 1496775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet * limitations under the License. 1596775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet */ 1696775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet 1796775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet#include "Manager.h" 1896775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet#include "NeuralNetworksWrapper.h" 1996775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet#include "Utils.h" 2096775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet 2196775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet#include <gtest/gtest.h> 2296775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet 2396775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouilletusing namespace android::nn::wrapper; 2496775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet 2596775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouilletint main(int argc, char** argv) { 2696775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet ::testing::InitGoogleTest(&argc, argv); 2796775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet 28820215d28bed6c90f696cde0f282445d16da432eMiao Wang android::nn::initVLogMask(); 29707dbd2d55f5dacf78ffb3ad7c8b3f37c2e9d758Jean-Luc Brouillet // Test with the installed drivers. 3096775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet int n1 = RUN_ALL_TESTS(); 3196775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet 32707dbd2d55f5dacf78ffb3ad7c8b3f37c2e9d758Jean-Luc Brouillet // Test with the CPU driver only. 33707dbd2d55f5dacf78ffb3ad7c8b3f37c2e9d758Jean-Luc Brouillet android::nn::DeviceManager::get()->setUseCpuOnly(true); 3496775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet int n2 = RUN_ALL_TESTS(); 35707dbd2d55f5dacf78ffb3ad7c8b3f37c2e9d758Jean-Luc Brouillet return n1 | n2; 3696775128e3bcfdc5be51b62edc50309c83861fe8Jean-Luc Brouillet} 37