Add Arm(R) Ethos(TM)-U85 NPU build option (#2701)

BUG=missing build support for Ethos-U85
tensorflow · Sep 26, 2024 · 722913e · 722913e
1 parent ef2179c
commit 722913e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tensorflow/lite/micro/tools/make/ext_libs/ethos_u.inc b/tensorflow/lite/micro/tools/make/ext_libs/ethos_u.inc
@@ -1,4 +1,4 @@
-# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2024 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -74,6 +74,10 @@ else ifeq ($(ETHOSU_ARCH), u65)
   ETHOSU_FLAGS += \
     -DETHOSU_ARCH=u65 \
     -DETHOSU65
+else ifeq ($(ETHOSU_ARCH), u85)
+  ETHOSU_FLAGS += \
+    -DETHOSU_ARCH=u85 \
+    -DETHOSU85
 else
   $(error "ETHOSU_ARCH=$(ETHOSU_ARCH) is not supported")
 endif