1 /* 2 * Copyright (C) 2019 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #define LOG_TAG "BurstBuilder" 18 19 #include "BurstBuilder.h" 20 21 #include <nnapi/IBurst.h> 22 23 #include <memory> 24 #include <utility> 25 #include <vector> 26 27 #include "CompilationBuilder.h" 28 29 namespace android { 30 namespace nn { 31 BurstBuilder(const CompilationBuilder * compilation,std::vector<SharedBurst> burstControllers)32BurstBuilder::BurstBuilder(const CompilationBuilder* compilation, 33 std::vector<SharedBurst> burstControllers) 34 : mCompilation(compilation), mBurstControllers(std::move(burstControllers)) {} 35 tryLock()36bool BurstBuilder::tryLock() { 37 const bool alreadyRunning = mCurrentlyRunning.test_and_set(); 38 return !alreadyRunning; 39 } 40 unlock()41void BurstBuilder::unlock() { 42 mCurrentlyRunning.clear(); 43 } 44 getCompilation() const45const CompilationBuilder* BurstBuilder::getCompilation() const { 46 return mCompilation; 47 } 48 getControllerAt(size_t index) const49SharedBurst BurstBuilder::getControllerAt(size_t index) const { 50 return index < mBurstControllers.size() ? mBurstControllers[index] : nullptr; 51 } 52 53 } // namespace nn 54 } // namespace android 55