From 58c8746e30254537423fe2dad06b67fd1cbbaa4a Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 16 Mar 2021 13:35:36 +0800
Subject: [PATCH] fix(opr): fix fast-run error in cuda

GitOrigin-RevId: 28dd187df99229d430e4b94861fd0da1fecd67f3
---
 dnn/src/common/algo_base.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dnn/src/common/algo_base.h b/dnn/src/common/algo_base.h
index e0f3202b..85771ab1 100644
--- a/dnn/src/common/algo_base.h
+++ b/dnn/src/common/algo_base.h
@@ -93,7 +93,7 @@ template <std::size_t I = 0, typename Opr, typename... Tp>
         I<sizeof...(Tp), void>::type set_sub_execution_policy(
                 const Opr* opr, std::tuple<Tp...>& t) {
     std::get<I>(t)->execution_policy() = opr->execution_policy().sub_policy[I];
-    set_sub_execution_policy<I + 1, Tp...>(opr, t);
+    set_sub_execution_policy<I + 1, Opr, Tp...>(opr, t);
 }
 
 template <typename Opr, typename... SubOpr>
@@ -103,7 +103,7 @@ void set_execution_policy(const Opr* opr, SubOpr... sub_oprs) {
         megdnn_assert(opr->execution_policy().sub_policy.size() ==
                       sizeof...(sub_oprs));
         auto&& sub = std::make_tuple(sub_oprs...);
-        set_sub_execution_policy<sizeof...(sub_oprs), Opr, SubOpr...>(opr, sub);
+        set_sub_execution_policy<0, Opr, SubOpr...>(opr, sub);
     }
 }
 
-- 
GitLab