diff --git a/src/opr/impl/search_policy/algo_chooser.cpp b/src/opr/impl/search_policy/algo_chooser.cpp index 71c4b2e7..f3695c82 100644 --- a/src/opr/impl/search_policy/algo_chooser.cpp +++ b/src/opr/impl/search_policy/algo_chooser.cpp @@ -853,7 +853,8 @@ AlgoChooser::AlgoChooserHelper::profile_single_algo( src.to_string().c_str()); param.dtypes[i] = src.dtype.enumv(); } - param.comp_node_loc = m_cn.locator(); + param.comp_node_physical = m_cn.locator(); + param.comp_node_logical = m_cn.locator_logical(); mgb_assert(param.shapes.size() == m_fastrun_layouts.size()); for (size_t i = 0; i < param.shapes.size(); ++i) param.shapes[i] = m_fastrun_layouts[i]; diff --git a/src/opr/impl/search_policy/profiler.cpp b/src/opr/impl/search_policy/profiler.cpp index 379c208e..9bc2f7b5 100644 --- a/src/opr/impl/search_policy/profiler.cpp +++ b/src/opr/impl/search_policy/profiler.cpp @@ -222,7 +222,8 @@ typename TimedProfiler::TResult TimedProfiler::prof_impl( mgb_assert(miopen_algo_search_enabled, "MIOpen algo search not enabled"); #endif auto&& param = raw_param.as_single_pod(); - CompNode cn = CompNode::load(param.comp_node_loc, param.comp_node_loc); + CompNode cn = + CompNode::load(param.comp_node_physical, param.comp_node_logical); auto megdnn_opr = intl::create_megdnn_opr(cn); std::array layouts; @@ -395,7 +396,8 @@ void TimedProfiler::prof_init_device(const TParam& raw_param) { megcore::enableMIOpenAlgoSearch(true); #endif auto&& param = raw_param.as_single_pod(); - CompNode cn = CompNode::load(param.comp_node_loc, param.comp_node_loc); + CompNode cn = + CompNode::load(param.comp_node_physical, param.comp_node_logical); // wait for cuda init, so its time does not get accounted in timeout cn.sync(); MIDOUT_E diff --git a/src/opr/include/megbrain/opr/search_policy/profiler.h b/src/opr/include/megbrain/opr/search_policy/profiler.h index ac8879ec..c81ee8b0 100644 --- a/src/opr/include/megbrain/opr/search_policy/profiler.h +++ b/src/opr/include/megbrain/opr/search_policy/profiler.h @@ -122,7 +122,7 @@ public: ExecutionPolicyBlob execution_policy; size_t workspace; megdnn::DTypeEnum dtypes[arity]; - CompNode::Locator comp_node_loc; + CompNode::Locator comp_node_physical, comp_node_logical; TensorShapeArray shapes; typename Opr::Param opr_param; bool allow_weight_preprocess;