fix multioutputs bug

PaddlePaddle · Nov 1, 2021 · 059cb03 · 059cb03 · paddle-bot-old · Nov 1, 2021
1 parent cf5cc45
commit 059cb03
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/paddle/pten/core/kernel_context.h b/paddle/pten/core/kernel_context.h
@@ -120,7 +120,7 @@ class KernelContext {
 
   template <typename TensorType>
   std::vector<TensorType*> MutableOutputAt(size_t start, size_t end) {
-    std::vector<TensorType> v;
+    std::vector<TensorType*> v;
     for (size_t i = start; i < end; ++i) {
       v.emplace_back(static_cast<TensorType*>(outputs_.at(i).get()));
     }

diff --git a/paddle/pten/core/kernel_utils.h b/paddle/pten/core/kernel_utils.h
@@ -146,15 +146,15 @@ using XPUContext = paddle::platform::XPUDeviceContext;
 
 #define PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(tensor_type)     \
   template <typename... Tail>                                            \
-  struct KernelCallHelper<const std::vector<tensor_type*>&, Tail...> {   \
+  struct KernelCallHelper<std::vector<tensor_type*>, Tail...> {          \
     template <int dev_ctx_idx,                                           \
               int in_idx,                                                \
               int attr_idx,                                              \
               int out_idx,                                               \
               typename... PreviousArgs>                                  \
     static void Compute(KernelContext* ctx, PreviousArgs&... pargs) {    \
       const std::pair<int, int> range = ctx->OutputRangeAt(out_idx);     \
-      const std::vector<tensor_type*> arg = std::move(                   \
+      std::vector<tensor_type*> arg = std::move(                         \
           ctx->MutableOutputAt<tensor_type>(range.first, range.second)); \
       KernelCallHelper<Tail...>::                                        \
           template Compute<dev_ctx_idx, in_idx, attr_idx, out_idx + 1>(  \