!14295 Clean codex
From: @jojobugfree Reviewed-by: @kisnwang,@jjfeing Signed-off-by: @jjfeingpull/14295/MERGE
commit
5787e5dcb3
@ -1,39 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#include "runtime/device/executor/executor_callback.h"
|
||||
#include "utils/log_adapter.h"
|
||||
|
||||
namespace mindspore {
|
||||
namespace device {
|
||||
void ExecutorCallback::RegistCallback(const std::function<void()> &callback) {
|
||||
std::lock_guard<std::mutex> guard(lock_);
|
||||
callback_queue_.push(callback);
|
||||
}
|
||||
|
||||
void ExecutorCallback::Consume() {
|
||||
std::lock_guard<std::mutex> guard(lock_);
|
||||
while (!callback_queue_.empty()) {
|
||||
auto callback_func = callback_queue_.front();
|
||||
callback_queue_.pop();
|
||||
if (!callback_func) {
|
||||
MS_LOG(EXCEPTION) << "callback_func is empty";
|
||||
}
|
||||
callback_func();
|
||||
}
|
||||
}
|
||||
} // namespace device
|
||||
} // namespace mindspore
|
@ -1,47 +0,0 @@
|
||||
/**
|
||||
* Copyright 2020 Huawei Technologies Co., Ltd
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_MINDSPORE_CCSRC_RUNTIME_DEVICE_EXECUTOR_EXECUTOR_CALLBACK_H_
|
||||
#define MINDSPORE_MINDSPORE_CCSRC_RUNTIME_DEVICE_EXECUTOR_EXECUTOR_CALLBACK_H_
|
||||
|
||||
#include <queue>
|
||||
#include <mutex>
|
||||
#include <functional>
|
||||
#include "utils/ms_utils.h"
|
||||
|
||||
namespace mindspore {
|
||||
namespace device {
|
||||
class ExecutorCallback {
|
||||
public:
|
||||
static ExecutorCallback &GetInstance() {
|
||||
static ExecutorCallback instance;
|
||||
return instance;
|
||||
}
|
||||
|
||||
void RegistCallback(const std::function<void()> &callback);
|
||||
void Consume();
|
||||
|
||||
private:
|
||||
ExecutorCallback() = default;
|
||||
~ExecutorCallback() = default;
|
||||
DISABLE_COPY_AND_ASSIGN(ExecutorCallback);
|
||||
|
||||
std::queue<std::function<void()>> callback_queue_;
|
||||
std::mutex lock_;
|
||||
};
|
||||
} // namespace device
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_MINDSPORE_CCSRC_RUNTIME_DEVICE_EXECUTOR_EXECUTOR_CALLBACK_H_
|
Loading…
Reference in new issue