Commit dab7172a by jason-song-dev Committed by Tianqi Chen

fix error when memory_id is VTA_MEM_ID_OUT (#4330)

parent 3486e2c2
...@@ -1016,7 +1016,7 @@ class CommandQueue { ...@@ -1016,7 +1016,7 @@ class CommandQueue {
elem_bytes = VTA_ACC_ELEM_BYTES; elem_bytes = VTA_ACC_ELEM_BYTES;
break; break;
case VTA_MEM_ID_OUT: case VTA_MEM_ID_OUT:
elem_bytes = VTA_INP_ELEM_BYTES; elem_bytes = VTA_OUT_ELEM_BYTES;
break; break;
default: default:
LOG(FATAL) << "Memory id not recognized:" << memory_id; LOG(FATAL) << "Memory id not recognized:" << memory_id;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment