p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixed bugs

This commit is contained in:
wangchangheng 2023-01-04 17:48:53 +08:00
parent 500a0d9c23
commit a913ef1b73
2 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/pipeline/pynative/grad/grad.cc
View File

@ -780,6 +780,7 @@ void GradExecutor::CheckNeedCompileGraph(const InputArgsInfoPtr &input_args_info
{ {
py::gil_scoped_acquire acquire; py::gil_scoped_acquire acquire;
EraseTopCellFromTopCellList(pre_top_cell); EraseTopCellFromTopCellList(pre_top_cell);
pre_top_cell->ClearDeviceMemory();
} }
already_run_top_cell_[already_top_cell_id] = new_top_cell; already_run_top_cell_[already_top_cell_id] = new_top_cell;
new_top_cell->set_force_top_cell_compile(false); new_top_cell->set_force_top_cell_compile(false);

11
mindspore/ccsrc/runtime/pynative/async/async_queue.cc
View File

@ -40,17 +40,10 @@ void AsyncQueue::WorkerLoop() {
while (true) { while (true) {
std::shared_ptr<AsyncTask> task; std::shared_ptr<AsyncTask> task;
bool task_empty = false;
{ {
std::unique_lock<std::mutex> lock(task_mutex_); std::unique_lock<std::mutex> lock(task_mutex_);
task_empty = tasks_.empty(); task_cond_var_.wait(lock, [this]() { return !tasks_.empty(); });
if (!task_empty) { task = tasks_.front();
task = tasks_.front();
}
}
if (task_empty) {
std::this_thread::yield();
continue;
} }
MS_LOG(DEBUG) << "Get task"; MS_LOG(DEBUG) << "Get task";