Skip to content

Commit ee51232

Browse files
committed
fix_typo
1 parent 3ad0db8 commit ee51232

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

paddle/fluid/eager/api/manual/eager_manual/forwards/dtensor_to_local_fwd_func.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ paddle::Tensor dtensor_to_local_ad_function(const paddle::Tensor& input) {
2323
VLOG(3) << "Running AD API: "
2424
<< "dtensor_to_local dygraph";
2525
bool rank_is_in_current_mesh = phi::distributed::IsCurRankInMesh(
26-
static_cast<phi::distributed::DistTensor*>(self->tensor.impl().get())
26+
static_cast<phi::distributed::DistTensor*>(input.impl().get())
2727
->process_mesh());
2828

2929
if (!rank_is_in_current_mesh) {

paddle/fluid/eager/api/manual/eager_manual/nodes/dtensor_to_local_node.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ DtensorToLocalGradNode::operator()(
3131
VLOG(3) << "Running AD API GRAD: "
3232
<< "dtensor_to_local";
3333
bool rank_is_in_current_mesh = phi::distributed::IsCurRankInMesh(
34-
static_cast<phi::distributed::DistTensor*>(self->tensor.impl().get())
34+
static_cast<phi::distributed::DistTensor*>(grads.impl().get())
3535
->process_mesh());
3636

3737
if (!rank_is_in_current_mesh) {

0 commit comments

Comments
 (0)