fix error when memory_id is VTA_MEM_ID_OUT (apache#4330)

zxy844288792 · Nov 15, 2019 · 16c60d0 · 16c60d0
1 parent 7021a21
commit 16c60d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vta/src/runtime.cc b/vta/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;