From b95eb38b8a797f3995162e7558e0bd2f0b22efd9 Mon Sep 17 00:00:00 2001
From: JamesLim <61349199+JamesLim-sy@users.noreply.github.com>
Date: Mon, 22 Feb 2021 08:55:49 +0800
Subject: [PATCH] fix the bug in backward OP of index_sample. (#31026)

---
 paddle/fluid/operators/index_sample_op.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddle/fluid/operators/index_sample_op.cu b/paddle/fluid/operators/index_sample_op.cu
index c8488eefb9..46dd91fed6 100644
--- a/paddle/fluid/operators/index_sample_op.cu
+++ b/paddle/fluid/operators/index_sample_op.cu
@@ -55,7 +55,7 @@ __global__ void IndexSampleGrad(const IndexT* index, T* in_grad,
       platform::CudaAtomicAdd(&(in_grad[in_idx - index_i + sample_idx]),
                               out_grad[sample_idx]);
     } else {
-      in_grad[in_idx - index_i + sample_idx] = out_grad[sample_idx];
+      in_grad[in_idx - index_i + sample_idx] = out_grad[index_idx];
     }
   }
 }
-- 
GitLab