commit 0388426ac1beb3fe9e3fe21479b60da61e71b639
author Lun Dong <lundong@google.com> Mon Jul 22 18:14:41 2024 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Mon Jul 22 18:14:41 2024 +0000
tree 59eaa2640adbb6af9fc8f5157ea6afb0c72e12df
parent 8556551877e0a0eb8b288abf20bd522de912d586
parent be6281cfb455a17062072b1e0afebb942cda1939

Merge "sw/kelvin: add test target for newly added mul op for Kelvin"

tests/tflm/BUILD

diff --git a/tests/tflm/BUILD b/tests/tflm/BUILD
index 183451f..3834394 100644
--- a/tests/tflm/BUILD
+++ b/tests/tflm/BUILD
@@ -102,6 +102,23 @@
 )
 
 kelvin_test(
+    name = "mul_test",
+    srcs = [
+        "@tflite-micro//tensorflow/lite/micro/kernels:mul_test.cc",
+    ],
+    hw_test_size = "large",
+    deps = [
+        "//crt",
+        "@tflite-micro//tensorflow/lite/c:common",
+        "@tflite-micro//tensorflow/lite/kernels/internal:tensor",
+        "@tflite-micro//tensorflow/lite/micro:micro_utils",
+        "@tflite-micro//tensorflow/lite/micro:test_helpers",
+        "@tflite-micro//tensorflow/lite/micro/kernels:kernel_runner",
+        "@tflite-micro//tensorflow/lite/micro/testing:micro_test",
+    ],
+)
+
+kelvin_test(
     name = "pooling_test",
     srcs = [
         "@tflite-micro//tensorflow/lite/micro/kernels:pooling_test.cc",