fix error when memory_id is VTA_MEM_ID_OUT (#4330)

dab7172a · jason-song-dev · Tianqi Chen · 3486e2c2 · dab7172a
Commit dab7172a authored Nov 14, 2019 by jason-song-dev Committed by Tianqi Chen Nov 13, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

vta/src/runtime.cc
+1 -1

No files found.
--- 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;