Skip to content

Commit

Permalink
fix for workspace allocation error
Browse files Browse the repository at this point in the history
  • Loading branch information
zhanghaohit committed Feb 5, 2021
1 parent d71a66f commit bbb5aed
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion src/runtime/workspace_pool.cc
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ class WorkspacePool::Pool {
}
// Release all resources
void Release(TVMContext ctx, DeviceAPI* device) {
ICHECK_EQ(allocated_.size(), 1);
// ICHECK_EQ(allocated_.size(), 1);
for (size_t i = 1; i < free_list_.size(); ++i) {
device->FreeDataSpace(ctx, free_list_[i].data);
}
Expand Down
2 changes: 1 addition & 1 deletion vta/tutorials/frontend/deploy_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -220,7 +220,7 @@
remote.upload(temp.relpath("graphlib.tar"))
lib = remote.load_module("graphlib.tar")

if env.TARGET == "sim" or env.TARGET == "intelfocl":
if env.TARGET == "intelfocl" or env.TARGET == "sim":
ctxes = [remote.ext_dev(0), remote.cpu(0)]
m = graph_runtime.create(graph, lib, ctxes)
else:
Expand Down

0 comments on commit bbb5aed

Please sign in to comment.