From 779d39da2ca9fa60daf78de3747eff902b20f270 Mon Sep 17 00:00:00 2001
From: jason-song-dev <37959160+jason-song-dev@users.noreply.github.com>
Date: Thu, 14 Nov 2019 12:42:22 +0900
Subject: [PATCH] fix error when memory_id is VTA_MEM_ID_OUT (#4330)

---
 src/runtime.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/runtime.cc b/src/runtime.cc
index 79fc0e293a85..b657d63a95f5 100644
--- a/src/runtime.cc
+++ b/src/runtime.cc
@@ -1016,7 +1016,7 @@ class CommandQueue {
           elem_bytes = VTA_ACC_ELEM_BYTES;
           break;
       case VTA_MEM_ID_OUT:
-          elem_bytes = VTA_INP_ELEM_BYTES;
+          elem_bytes = VTA_OUT_ELEM_BYTES;
           break;
       default:
           LOG(FATAL) << "Memory id not recognized:" << memory_id;