Skip to content

Commit 39a0d0a

Browse files
authored
add no_need_buffer into yaml for view op. (#73815)
1 parent a926cd3 commit 39a0d0a

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

paddle/phi/ops/yaml/backward.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3859,6 +3859,7 @@
38593859
kernel :
38603860
func : view_dtype_grad
38613861
data_type : out_grad
3862+
no_need_buffer: input
38623863

38633864
- backward_op : warpctc_grad
38643865
forward : warpctc (Tensor logits, Tensor label, Tensor logits_length, Tensor labels_length, int blank = 0, bool norm_by_times = false) -> Tensor(loss), Tensor(warpctcgrad)

paddle/phi/ops/yaml/inconsistent/dygraph_backward.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,7 @@
365365
args : (Tensor input, Tensor out_grad, int64_t[] dims = {})
366366
output : Tensor(input_grad)
367367
invoke: view_shape(out_grad, input.shape())
368+
no_need_buffer: input
368369

369370
- backward_op: fused_gemm_epilogue_grad
370371
forward : fused_gemm_epilogue(Tensor x, Tensor y, Tensor bias, bool trans_x, bool trans_y, str activation) -> Tensor(out), Tensor(reserve_space)

0 commit comments

Comments
 (0)