Merge pull request !6157 from gukecai/codex
pull/6157/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit d96831bfa7

@ -53,7 +53,6 @@ class AscendProfiler {
std::vector<Event> events_;
std::atomic_int counter_;
};
} // namespace ascend
} // namespace profiler
} // namespace mindspore

@ -26,7 +26,6 @@
namespace mindspore {
namespace profiler {
namespace ascend {
inline pid_t GetTid() {
thread_local static pid_t tid = syscall(__NR_gettid);
return tid;

@ -21,7 +21,6 @@
namespace mindspore {
namespace profiler {
namespace ascend {
CallbackManager::CallbackManager(rtStream_t stream) : stream_(stream) {}
Status CallbackManager::Init() {

@ -123,7 +123,9 @@ class AscendStreamAssign {
void CheckEventAssign(const NotNull<KernelGraphPtr> &graph_ptr);
void AssignAllNodesStream(const NotNull<KernelGraphPtr> &graph_ptr);
void AssignCommonStreamId(const CNodePtr &cur_cnode_ptr);
void AssignHcom(const NotNull<KernelGraphPtr> &graph_ptr);
uint32_t AssignHcomStreamId(const CNodePtr &cur_cnode_ptr, bool new_graph);
void AssignIndependent(const NotNull<KernelGraphPtr> &graph_ptr);
uint32_t AssignIndependentStreamId(const CNodePtr &cur_cnode_ptr, bool new_graph);
void UpdateAtomicAddrCleanStreamId(const NotNull<KernelGraphPtr> &graph_ptr);
void FindHcomParallelStreams(const NotNull<KernelGraphPtr> &graph_ptr);
@ -137,6 +139,7 @@ class AscendStreamAssign {
std::map<uint32_t, std::set<uint32_t>> other_graph);
void UpdateStreamSwitch(const NotNull<KernelGraphPtr> &graph_ptr, const CNodePtr &switch_ptr,
vector<CNodePtr> *orders);
bool CheckStreamSwitch(const CNodePtr &switch_ptr);
void InsertEventForIndependentParallel(const NotNull<KernelGraphPtr> &graph_ptr);
void InsertCtrlForIndependentParallel(const NotNull<KernelGraphPtr> &graph_ptr);
void InsertEventForHcomParallel(const NotNull<KernelGraphPtr> &graph_ptr);

Loading…
Cancel
Save