提交 ab698834 编写于 作者: M Megvii Engine Team

fix(mgb/core): use thread local for OpenCL recorder

GitOrigin-RevId: 76da0c2a24ca944a327cf1b2010aa91ffa512cb4
上级 88574565
......@@ -106,13 +106,11 @@ class CpuCompNode::SeqRecorderImpl final : public CompNodeSeqRecorder {
std::vector<TaskElem> m_tasks;
ThreadPool* m_thread_pool = nullptr;
const CompNode m_record_compnode;
/*!
* \brief use to check the all ther recording tasks are its self CompNode
* related task, void hook other CompNode related task to the recorder.
*/
void check_the_same_comp_node(const CompNode& comp_node) const;
public:
SeqRecorderImpl(SeqRecorderImpl** self_pointer, ThreadPool* thread_pool,
const CompNode& comp_node)
......
......@@ -577,6 +577,7 @@ public:
virtual void exit_fake_exec(const CompNode& comp_node) = 0;
virtual void stop(const CompNode& comp_node) = 0;
virtual void replay() = 0;
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册