Paddle transformer gpu ce fails

ou6hu8tu  于 2022-10-20  发布在  其他
关注(0)|答案(1)|浏览(162)

due to #17080

kxe2p93d

kxe2p93d1#

If I add following codes in PrepareData between line 1025 and 1026
Paddle/paddle/fluid/framework/operator.cc

Lines 1025 to 1026 in 5e5e7b3

| | Scope* new_scope = nullptr; |
| | if (!need_prepare_data_) return new_scope; |

if (type_ == "gather")
   need_prepare_data_ = true;

This problem is fixed.
@sneaxiy Is this problem related to DECLARE_NO_NEED_BUFFER_VARS_INFERENCE ?

Paddle/paddle/fluid/operators/gather_op.cc

Line 119 in 5e5e7b3

| | DECLARE_NO_NEED_BUFFER_VARS_INFERENCE(GatherGradNoNeedBufferVarInference, "X"); |

相关问题