1 files changed, 97 insertions, 0 deletions
diff --git a/include/ethosu_driver.h b/include/ethosu_driver.h
new file mode 100644
index 0000000..5d75852
--- /dev/null
+++ b/include/ethosu_driver.h
@@ -0,0 +1,97 @@
+/*
+ * Copyright (c) 2019-2020 Arm Limited. All rights reserved.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * 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
+ *
+ * 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.
+ */
+
+#pragma once
+
+/******************************************************************************
+ * Includes
+ ******************************************************************************/
+
+#include <stdint.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+/******************************************************************************
+ * Defines
+ ******************************************************************************/
+
+/******************************************************************************
+ * Types
+ ******************************************************************************/
+
+struct ethosu_version_id
+{
+    // Ethos-U id
+    uint8_t version_status;
+    uint8_t version_minor;
+    uint8_t version_major;
+    uint8_t product_major;
+    uint8_t arch_patch_rev;
+    uint8_t arch_minor_rev;
+    uint8_t arch_major_rev;
+
+    // Driver Version
+    uint8_t driver_patch_rev;
+    uint8_t driver_minor_rev;
+    uint8_t driver_major_rev;
+};
+
+struct ethosu_version_config
+{
+    uint8_t macs_per_cc;
+    uint8_t cmd_stream_version;
+    uint8_t shram_size;
+};
+
+struct ethosu_version
+{
+    struct ethosu_version_id id;
+    struct ethosu_version_config cfg;
+};
+
+/******************************************************************************
+ * Prototypes
+ ******************************************************************************/
+
+/**
+ * Initialize the Ethos-U driver.
+ */
+int ethosu_init(void);
+
+/**
+ * Get Ethos-U version.
+ */
+int ethosu_get_version(struct ethosu_version *version);
+
+/**
+ * Invoke Vela command stream.
+ */
+int ethosu_invoke(const void *custom_data_ptr,
+                  const int custom_data_size,
+                  const uint64_t *base_addr,
+                  const int num_base_addr);
+/**
+ * Abort Ethos-U inference.
+ */
+void ethosu_abort(void);
+
+#ifdef __cplusplus
+}
+#endif