From 265e74ff1668d21214c67c360c0bef91c44ca7fb Mon Sep 17 00:00:00 2001
From: xie-oritek <142767785+xie-oritek@users.noreply.github.com>
Date: Tue, 22 Aug 2023 17:37:38 +0800
Subject: [PATCH] Add int4/uint4 definition (#638)

---
 include/tim/vx/types.h   | 2 ++
 src/tim/vx/type_utils.cc | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/include/tim/vx/types.h b/include/tim/vx/types.h
index 1937e68..432ff1c 100644
--- a/include/tim/vx/types.h
+++ b/include/tim/vx/types.h
@@ -29,6 +29,8 @@ namespace vx {
 
 enum class DataType {
   UNKNOWN,
+  INT4,
+  UINT4,
   INT8,
   UINT8,
   INT16,
diff --git a/src/tim/vx/type_utils.cc b/src/tim/vx/type_utils.cc
index 11d5681..a701452 100644
--- a/src/tim/vx/type_utils.cc
+++ b/src/tim/vx/type_utils.cc
@@ -27,6 +27,10 @@ namespace tim {
 namespace vx {
 vsi_nn_type_e TranslateDataType(DataType dtype) {
   switch (dtype) {
+    case DataType::INT4:
+        return VSI_NN_TYPE_INT4;
+    case DataType::UINT4:
+        return VSI_NN_TYPE_UINT4;
     case DataType::INT8:
       return VSI_NN_TYPE_INT8;
     case DataType::UINT8: