| /* |
| * Copyright (C) 2019 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package android.hardware.neuralnetworks@1.3; |
| |
| import @1.1::ExecutionPreference; |
| import @1.2::Constant; |
| import @1.2::DeviceType; |
| import @1.2::Extension; |
| import @1.2::IDevice; |
| import BufferDesc; |
| import BufferRole; |
| import Capabilities; |
| import ErrorStatus; |
| import Model; |
| import OptionalTimePoint; |
| import Priority; |
| import IBuffer; |
| import IPreparedModel; |
| import IPreparedModelCallback; |
| |
| /** |
| * This interface represents a device driver. |
| */ |
| interface IDevice extends @1.2::IDevice { |
| /** |
| * Gets the capabilities of a driver. |
| * |
| * @return status Error status of the call, must be: |
| * - NONE if successful |
| * - DEVICE_UNAVAILABLE if driver is offline or busy |
| * - GENERAL_FAILURE if there is an unspecified error |
| * @return capabilities Capabilities of the driver. |
| */ |
| getCapabilities_1_3() generates (ErrorStatus status, Capabilities capabilities); |
| |
| /** |
| * Gets the supported operations in a model. |
| * |
| * getSupportedOperations indicates which operations of the top-level |
| * subgraph are fully supported by the vendor driver. If an operation may |
| * not be supported for any reason, getSupportedOperations must return |
| * false for that operation. |
| * |
| * The {@link OperationType::IF} and {@link OperationType::WHILE} |
| * operations may only be fully supported if the vendor driver fully |
| * supports all operations in the referenced subgraphs. |
| * |
| * @param model A model whose operations--and their corresponding operands-- |
| * are to be verified by the driver. |
| * @return status Error status of the call, must be: |
| * - NONE if successful |
| * - DEVICE_UNAVAILABLE if driver is offline or busy |
| * - GENERAL_FAILURE if there is an unspecified error |
| * - INVALID_ARGUMENT if provided model is invalid |
| * @return supportedOperations A list of supported operations, where true |
| * indicates the operation is supported and false indicates the |
| * operation is not supported. The index of "supported" corresponds with |
| * the index of the operation it is describing. |
| */ |
| getSupportedOperations_1_3(Model model) |
| generates (ErrorStatus status, vec<bool> supportedOperations); |
| |
| /** |
| * Asynchronously creates a prepared model for execution and optionally |
| * saves it into cache files. |
| * |
| * prepareModel is used to make any necessary transformations to or |
| * alternative representations to a model for execution, possibly including |
| * transformations on the constant data, optimization on the model's graph, |
| * or compilation into the device's native binary format. The model itself |
| * is not changed. |
| * |
| * Optionally, caching information may be provided for the driver to save |
| * the prepared model to cache files for faster model compilation time when |
| * the same model preparation is requested in the future. There are two |
| * types of cache file handles provided to the driver: model cache and data |
| * cache. For more information on the two types of cache handles, refer to |
| * getNumberOfCacheFilesNeeded. |
| * |
| * The file descriptors must be opened with read and write permission. A |
| * file may have any size, and the corresponding file descriptor may have |
| * any offset. The driver must truncate a file to zero size before writing |
| * to that file. The file descriptors may be closed by the client once the |
| * asynchronous preparation has finished. The driver must dup a file |
| * descriptor if it wants to get access to the cache file later. |
| * |
| * The model is prepared asynchronously with respect to the caller. The |
| * prepareModel function must verify the inputs to the preparedModel |
| * function related to preparing the model (as opposed to saving the |
| * prepared model to cache) are correct. If there is an error, prepareModel |
| * must immediately invoke the callback with the appropriate ErrorStatus |
| * value and nullptr for the IPreparedModel, then return with the same |
| * ErrorStatus. If the inputs to the prepareModel function that are related |
| * to preparing the model are valid and there is no error, prepareModel must |
| * launch an asynchronous task to prepare the model in the background, and |
| * immediately return from prepareModel with ErrorStatus::NONE. If the |
| * asynchronous task fails to launch, prepareModel must immediately invoke |
| * the callback with ErrorStatus::GENERAL_FAILURE and nullptr for the |
| * IPreparedModel, then return with ErrorStatus::GENERAL_FAILURE. |
| * |
| * When the asynchronous task has finished preparing the model, it must |
| * immediately invoke the callback function provided as an input to |
| * prepareModel. If the model was prepared successfully, the callback object |
| * must be invoked with an error status of ErrorStatus::NONE and the |
| * produced IPreparedModel object. If an error occurred preparing the model, |
| * the callback object must be invoked with the appropriate ErrorStatus |
| * value and nullptr for the IPreparedModel. |
| * |
| * The model is prepared with a priority. This priority is relative to other |
| * prepared models owned by the same client. Higher priority executions may |
| * use more compute resources than lower priority executions, and may |
| * preempt or starve lower priority executions. |
| * |
| * prepareModel_1_3 can be called with an optional deadline. If the model |
| * is not able to be prepared before the provided deadline, the model |
| * preparation may be aborted, and either {@link |
| * ErrorStatus::MISSED_DEADLINE_TRANSIENT} or {@link |
| * ErrorStatus::MISSED_DEADLINE_PERSISTENT} may be returned. The error due |
| * to an abort must be sent the same way as other errors, described above. |
| * The deadline is represented as nanoseconds since the epoch of the steady |
| * clock (as if from std::chrono::steady_clock::time_point), but the service |
| * may convert it to the nanoseconds since boot time (as if from |
| * clock_gettime(CLOCK_BOOTTIME, &ts) or |
| * android::base::boot_clock::time_point) to account for time when the |
| * system is suspended. This conversion can by done by finding the timeout |
| * duration remaining compared to the steady_clock and adding it to the |
| * current boot_clock time. |
| * |
| * Optionally, the driver may save the prepared model to cache during the |
| * asynchronous preparation. Any error that occurs when saving to cache must |
| * not affect the status of preparing the model. Even if the input arguments |
| * related to the cache may be invalid, or the driver may fail to save to |
| * cache, the prepareModel function must finish preparing the model. The |
| * driver may choose not to save to cache even if the caching information is |
| * provided and valid. |
| * |
| * The only information that may be unknown to the model at this stage is |
| * the shape of the tensors, which may only be known at execution time. As |
| * such, some driver services may return partially prepared models, where |
| * the prepared model may only be finished when it is paired with a set of |
| * inputs to the model. Note that the same prepared model object may be used |
| * with different shapes of inputs on different (possibly concurrent) |
| * executions. |
| * |
| * Multiple threads may call prepareModel on the same model concurrently. |
| * |
| * @param model The model to be prepared for execution. |
| * @param preference Indicates the intended execution behavior of a prepared |
| * model. |
| * @param priority The priority of the prepared model relative to other |
| * prepared models owned by the client. |
| * @param deadline The time by which the model is expected to be prepared. |
| * If the model cannot be prepared by the deadline, the preparation may |
| * be aborted. |
| * @param modelCache A vector of handles with each entry holding exactly one |
| * cache file descriptor for the security-sensitive cache. The length of |
| * the vector must either be 0 indicating that caching information is |
| * not provided, or match the numModelCache returned from |
| * getNumberOfCacheFilesNeeded. The cache handles will be provided in |
| * the same order when retrieving the preparedModel from cache files |
| * with prepareModelFromCache_1_3. |
| * @param dataCache A vector of handles with each entry holding exactly one |
| * cache file descriptor for the constants' cache. The length of the |
| * vector must either be 0 indicating that caching information is not |
| * provided, or match the numDataCache returned from |
| * getNumberOfCacheFilesNeeded. The cache handles will be provided in |
| * the same order when retrieving the preparedModel from cache files |
| * with prepareModelFromCache_1_3. |
| * @param token A caching token of length Constant::BYTE_SIZE_OF_CACHE_TOKEN |
| * identifying the prepared model. The same token will be provided when |
| * retrieving the prepared model from the cache files with |
| * prepareModelFromCache_1_3. Tokens should be chosen to have a low rate of |
| * collision for a particular application. The driver cannot detect a |
| * collision; a collision will result in a failed execution or in a |
| * successful execution that produces incorrect output values. If both |
| * modelCache and dataCache are empty indicating that caching |
| * information is not provided, this token must be ignored. |
| * @param callback A callback object used to return the error status of |
| * preparing the model for execution and the prepared model if |
| * successful, nullptr otherwise. The callback object's notify function |
| * must be called exactly once, even if the model could not be prepared. |
| * @return status Error status of launching a task which prepares the model |
| * in the background; must be: |
| * - NONE if preparation task is successfully launched |
| * - DEVICE_UNAVAILABLE if driver is offline or busy |
| * - GENERAL_FAILURE if there is an unspecified error |
| * - INVALID_ARGUMENT if one of the input arguments related to preparing |
| * the model is invalid |
| * - MISSED_DEADLINE_* if the preparation is aborted because the model |
| * cannot be prepared by the deadline |
| * - RESOURCE_EXHAUSTED_* if the task was aborted by the driver |
| */ |
| prepareModel_1_3(Model model, ExecutionPreference preference, |
| Priority priority, OptionalTimePoint deadline, |
| vec<handle> modelCache, vec<handle> dataCache, |
| uint8_t[Constant:BYTE_SIZE_OF_CACHE_TOKEN] token, |
| IPreparedModelCallback callback) |
| generates (ErrorStatus status); |
| |
| /** |
| * Creates a prepared model from cache files for execution. |
| * |
| * prepareModelFromCache_1_3 is used to retrieve a prepared model directly from |
| * cache files to avoid slow model compilation time. There are |
| * two types of cache file handles provided to the driver: model cache |
| * and data cache. For more information on the two types of cache handles, |
| * refer to getNumberOfCacheFilesNeeded. |
| * |
| * The file descriptors must be opened with read and write permission. A file may |
| * have any size, and the corresponding file descriptor may have any offset. The |
| * driver must truncate a file to zero size before writing to that file. The file |
| * descriptors may be closed by the client once the asynchronous preparation has |
| * finished. The driver must dup a file descriptor if it wants to get access to |
| * the cache file later. |
| * |
| * The model is prepared asynchronously with respect to the caller. The |
| * prepareModelFromCache_1_3 function must verify the inputs to the |
| * prepareModelFromCache_1_3 function are correct, and that the security-sensitive |
| * cache has not been modified since it was last written by the driver. |
| * If there is an error, or if compilation caching is not supported, or if the |
| * security-sensitive cache has been modified, prepareModelFromCache_1_3 must |
| * immediately invoke the callback with the appropriate ErrorStatus value and |
| * nullptr for the IPreparedModel, then return with the same ErrorStatus. If |
| * the inputs to the prepareModelFromCache_1_3 function are valid, the security-sensitive |
| * cache is not modified, and there is no error, prepareModelFromCache_1_3 must launch an |
| * asynchronous task to prepare the model in the background, and immediately return |
| * from prepareModelFromCache_1_3 with ErrorStatus::NONE. If the asynchronous task |
| * fails to launch, prepareModelFromCache_1_3 must immediately invoke the callback |
| * with ErrorStatus::GENERAL_FAILURE and nullptr for the IPreparedModel, then |
| * return with ErrorStatus::GENERAL_FAILURE. |
| * |
| * When the asynchronous task has finished preparing the model, it must |
| * immediately invoke the callback function provided as an input to |
| * prepareModelFromCache_1_3. If the model was prepared successfully, the |
| * callback object must be invoked with an error status of ErrorStatus::NONE |
| * and the produced IPreparedModel object. If an error occurred preparing |
| * the model, the callback object must be invoked with the appropriate |
| * ErrorStatus value and nullptr for the IPreparedModel. |
| * |
| * prepareModelFromCache_1_3 can be called with an optional deadline. If the |
| * model is not able to prepared before the provided deadline, the model |
| * preparation may be aborted, and either {@link |
| * ErrorStatus::MISSED_DEADLINE_TRANSIENT} |
| * or {@link ErrorStatus::MISSED_DEADLINE_PERSISTENT} may be returned. The |
| * error due to an abort must be sent the same way as other errors, |
| * described above. The deadline is represented as nanoseconds since the |
| * epoch of the steady clock (as if from |
| * std::chrono::steady_clock::time_point), but the service may convert it to |
| * the nanoseconds since boot time (as if from |
| * clock_gettime(CLOCK_BOOTTIME, &ts) or |
| * android::base::boot_clock::time_point) to account for time when the |
| * system is suspended. This conversion can by done by finding the timeout |
| * duration remaining compared to the steady_clock and adding it to the |
| * current boot_clock time. |
| * |
| * The only information that may be unknown to the model at this stage is |
| * the shape of the tensors, which may only be known at execution time. As |
| * such, some driver services may return partially prepared models, where |
| * the prepared model may only be finished when it is paired with a set of |
| * inputs to the model. Note that the same prepared model object may be |
| * used with different shapes of inputs on different (possibly concurrent) |
| * executions. |
| * |
| * @param deadline The time by which the model is expected to be prepared. |
| * If the model cannot be prepared by the deadline, the preparation may |
| * be aborted. |
| * @param modelCache A vector of handles with each entry holding exactly one |
| * cache file descriptor for the security-sensitive cache. The length of |
| * the vector must match the numModelCache returned from getNumberOfCacheFilesNeeded. |
| * The cache handles will be provided in the same order as with prepareModel_1_3. |
| * @param dataCache A vector of handles with each entry holding exactly one |
| * cache file descriptor for the constants' cache. The length of the vector |
| * must match the numDataCache returned from getNumberOfCacheFilesNeeded. |
| * The cache handles will be provided in the same order as with prepareModel_1_3. |
| * @param token A caching token of length Constant::BYTE_SIZE_OF_CACHE_TOKEN |
| * identifying the prepared model. It is the same token provided when saving |
| * the cache files with prepareModel_1_3. Tokens should be chosen |
| * to have a low rate of collision for a particular application. The driver |
| * cannot detect a collision; a collision will result in a failed execution |
| * or in a successful execution that produces incorrect output values. |
| * @param callback A callback object used to return the error status of |
| * preparing the model for execution and the prepared model if |
| * successful, nullptr otherwise. The callback object's notify function |
| * must be called exactly once, even if the model could not be prepared. |
| * @return status Error status of launching a task which prepares the model |
| * in the background; must be: |
| * - NONE if preparation task is successfully launched |
| * - DEVICE_UNAVAILABLE if driver is offline or busy |
| * - GENERAL_FAILURE if caching is not supported or if there is an |
| * unspecified error |
| * - INVALID_ARGUMENT if one of the input arguments is invalid |
| * - MISSED_DEADLINE_* if the preparation is aborted because the model |
| * cannot be prepared by the deadline |
| * - RESOURCE_EXHAUSTED_* if the task was aborted by the driver |
| */ |
| prepareModelFromCache_1_3(OptionalTimePoint deadline, |
| vec<handle> modelCache, vec<handle> dataCache, |
| uint8_t[Constant:BYTE_SIZE_OF_CACHE_TOKEN] token, |
| IPreparedModelCallback callback) |
| generates (ErrorStatus status); |
| |
| /** |
| * Allocates a driver-managed buffer with the properties specified by the buffer descriptor |
| * as well as the input and output roles. |
| * |
| * The allocate function must verify its inputs are correct. If there is an error, or if a |
| * certain role or property is not supported by the driver, the allocate |
| * function must return with an appropriate ErrorStatus, a nullptr as the IBuffer, and 0 as the |
| * buffer token. If the allocation is successful, this method must return with ErrorStatus::NONE |
| * and the produced IBuffer with a positive token identifying the allocated buffer. A successful |
| * allocation must accommodate all of the specified roles and buffer properties. |
| * |
| * The buffer is allocated to an uninitialized state. An uninitialized buffer may only be used |
| * in ways that are specified by outputRoles. A buffer is initialized after it is used as an |
| * output in a successful execution, or after a successful invocation of IBuffer::copyFrom on |
| * the buffer. An initialized buffer may be used according to all roles specified in inputRoles |
| * and outputRoles. A buffer will return to the uninitialized state if it is used as an output |
| * in a failed execution, or after a failed invocation of IBuffer::copyFrom on the buffer. |
| * |
| * The dimensions of the buffer can be deduced from the buffer descriptor as well as the |
| * dimensions of the corresponding model operands of the input and output roles. The dimensions |
| * or rank of the buffer may be unknown at this stage. As such, some driver services may only |
| * create a placeholder and defer the actual allocation until execution time. Note that the |
| * same buffer may be used for different shapes of outputs on different executions. When the |
| * buffer is used as an input, the input shape must be the same as the output shape from the |
| * last execution using this buffer as an output. |
| * |
| * The driver must apply proper validatation upon every usage of the buffer, and must fail the |
| * execution immediately if the usage is illegal. |
| * |
| * @param desc A buffer descriptor specifying the properties of the buffer to allocate. |
| * @param preparedModels A vector of IPreparedModel objects. Must only contain IPreparedModel |
| * objects from the same IDevice as this method is being invoked on. |
| * @param inputRoles A vector of roles with each specifying an input to a prepared model. |
| * @param outputRoles A vector of roles with each specifying an output to a prepared model. |
| * Each role specified in inputRoles and outputRoles must be unique. The corresponding |
| * model operands of the roles must have the same OperandType, scale, zero point, and |
| * ExtraParams. The dimensions of the operands and the dimensions specified in the buffer |
| * descriptor must be compatible with each other. Two dimensions are incompatible if there |
| * is at least one axis that is fully specified in both but has different values. |
| * @return status Error status of the buffer allocation. Must be: |
| * - NONE if successful |
| * - DEVICE_UNAVAILABLE if driver is offline or busy |
| * - GENERAL_FAILURE if a certain buffer property or a certain role is not supported, |
| * or if there is an unspecified error |
| * - INVALID_ARGUMENT if one of the input arguments is invalid |
| * @return buffer The allocated IBuffer object. If the buffer was unable to be allocated |
| * due to an error, nullptr must be returned. |
| * @return token A positive token identifying the allocated buffer. The same token will be |
| * provided when referencing the buffer as one of the memory pools in the request of an |
| * execution. The token must not collide with the tokens of other IBuffer objects that are |
| * currently alive in the same driver service. If the buffer was unable to be allocated |
| * due to an error, the token must be 0. |
| */ |
| allocate(BufferDesc desc, vec<IPreparedModel> preparedModels, vec<BufferRole> inputRoles, |
| vec<BufferRole> outputRoles) |
| generates (ErrorStatus status, IBuffer buffer, uint32_t token); |
| }; |