Skip to content

SeetaResearch / Dragon

Go to a project
Commit 53d5742d by Ting PAN
Options

add DragonBoard

1 parent 93943fc8
Inline Side-by-side
Showing with 7 additions and 3 deletions
.gitmodules 0 → 100644
[submodule "DragonBoard"]
path = DragonBoard
url = git://github.com/neopenx/DragonBoard.git
Dragon/modules/python/dragon.cc
...@@ -168,8 +168,8 @@ PyObject* ResetWorkspaceCC(PyObject* self, PyObject* args) { ...@@ -168,8 +168,8 @@ PyObject* ResetWorkspaceCC(PyObject* self, PyObject* args) {
return nullptr; return nullptr;
} }
LOG(INFO) << "Reset the Workspace(" << g_current_workspace << ")"; LOG(INFO) << "Reset the Workspace(" << g_current_workspace << ")";
string workspace_name = string(cname); string workspace_name = string(cname);
if (workspace_name.empty()) g_workspaces[g_current_workspace].reset(new Workspace()); if (workspace_name.empty()) g_workspaces[g_current_workspace].reset(new Workspace());
else g_workspaces[g_current_workspace].reset(new Workspace(workspace_name)); else g_workspaces[g_current_workspace].reset(new Workspace(workspace_name));
g_workspace = g_workspaces[g_current_workspace].get(); g_workspace = g_workspaces[g_current_workspace].get();
Py_RETURN_TRUE; Py_RETURN_TRUE;
...@@ -281,7 +281,7 @@ PyObject* FetchTensorCC(PyObject* self, PyObject* args) { ...@@ -281,7 +281,7 @@ PyObject* FetchTensorCC(PyObject* self, PyObject* args) {
TypeId type_id = CTypeToFetcher(tensor->meta().id()); TypeId type_id = CTypeToFetcher(tensor->meta().id());
CHECK(type_id != 0) << "\nTensor(" << tensor->name() CHECK(type_id != 0) << "\nTensor(" << tensor->name()
<< ") has not been computed yet."; << ") has not been computed yet.";
unique_ptr<TensorFetcherBase> fetcher(createFetcher(type_id)); unique_ptr<TensorFetcherBase> fetcher(CreateFetcher(type_id));
if (fetcher.get()) { if (fetcher.get()) {
return fetcher->Fetch(*tensor); // copy the tensor data to a numpy object return fetcher->Fetch(*tensor); // copy the tensor data to a numpy object
} else { } else {
......
DragonBoard @ c4e649f6
Subproject commit c4e649f66e59b39d07a1a857c72e721ebbdb93ae
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!