Skip to content

Commit 4a071e2

Browse files
【dy2st】modify tanspose api case (#68688)
1 parent 159aa58 commit 4a071e2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/pybind/pir.cc

+3-1
Original file line numberDiff line numberDiff line change
@@ -1856,7 +1856,9 @@ SplitedResult SplitForwardBackward(
18561856
backward_range);
18571857

18581858
pir::Block &backward_block = *backward_program->block();
1859-
bool has_backward = (backward_range[1] > backward_range[0]);
1859+
bool has_backward = forward_inputs_grads.size() > 0 ||
1860+
forward_params_grads.size() > 0 ||
1861+
forward_outputs_grads.size() > 0;
18601862

18611863
// forward program construct.
18621864
VLOG(4) << "start create forward program.";

0 commit comments

Comments
 (0)