Skip to content

Commit 4430ca2

Browse files
committed
Revert "[cherry-pick] This PR enable offset of generator for custom device. (PaddlePaddle#60616) (PaddlePaddle#60772)"
This reverts commit 0f732a5.
1 parent c9c6a5e commit 4430ca2

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

paddle/phi/core/generator.cc

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -278,8 +278,7 @@ uint64_t Generator::Random64() {
278278

279279
std::pair<uint64_t, uint64_t> Generator::IncrementOffset(
280280
uint64_t increment_offset) {
281-
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) || \
282-
defined(PADDLE_WITH_CUSTOM_DEVICE)
281+
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
283282
std::lock_guard<std::mutex> lock(this->mu_);
284283
uint64_t cur_offset = this->state_.thread_offset;
285284
VLOG(10) << "cur_offset = " << cur_offset

0 commit comments

Comments
 (0)