Merge pull request #95 from ConnorChristie/patch-1

Add C extern to support C++
2020-04-16 22:32:55 +08:00 · 2020-04-16 22:32:55 +08:00 · 196b8d3faa
parent 3f3d4055dd c236dc8508
commit 196b8d3faa
1 changed files with 8 additions and 0 deletions
--- a/lib/drivers/include/kpu.h
+++ b/lib/drivers/include/kpu.h
@ -19,6 +19,10 @@
 #include <stdint.h>
 #include "dmac.h"

+#ifdef __cplusplus
+extern "C" {
+#endif
+
 #define kpu_matmul_begin kpu_conv2d_output
 #define IOMEM 0x40000000

@ -960,4 +964,8 @@ int kpu_get_output(kpu_model_context_t *ctx, uint32_t index, uint8_t **data, siz
 */
 int kpu_run_kmodel(kpu_model_context_t *ctx, const uint8_t *src, dmac_channel_number_t dma_ch, kpu_done_callback_t done_callback, void *userdata);

+#ifdef __cplusplus
+}
+#endif
+
 #endif