Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 19c2871

Browse files
committedApr 9, 2024
change lock
1 parent 6cdfa4b commit 19c2871

File tree

1 file changed

+7
-11
lines changed

1 file changed

+7
-11
lines changed
 

‎dipu/torch_dipu/csrc_dipu/diopirt/diopirt_impl.cpp

+7-11
Original file line numberDiff line numberDiff line change
@@ -187,11 +187,9 @@ DIOPI_RT_API diopiError_t diopiGeneratorGetSeedAndOffset(
187187
diopiGeneratorHandle_t th, uint64_t& seed, uint64_t& offset) {
188188
auto generator = reinterpret_cast<at::Generator*>(th);
189189
auto gen_impl = at::check_generator<dipu::DIPUGeneratorImpl>(*generator);
190-
{
191-
std::lock_guard<std::mutex> lock(gen_impl->mutex_);
192-
offset = gen_impl->get_offset();
193-
seed = gen_impl->current_seed();
194-
}
190+
191+
offset = gen_impl->get_offset();
192+
seed = gen_impl->current_seed();
195193

196194
return diopiSuccess;
197195
}
@@ -200,12 +198,10 @@ DIOPI_RT_API diopiError_t diopiGeneratorSetSeedAndOffset(
200198
diopiGeneratorHandle_t th, uint64_t seed, uint64_t offset) {
201199
auto generator = reinterpret_cast<at::Generator*>(th);
202200
auto gen_impl = at::check_generator<dipu::DIPUGeneratorImpl>(*generator);
203-
{
204-
std::lock_guard<std::mutex> lock(gen_impl->mutex_);
205-
gen_impl->set_offset(offset);
206-
gen_impl->set_current_seed(seed);
207-
}
208-
201+
202+
gen_impl->set_offset(offset);
203+
gen_impl->set_current_seed(seed);
204+
209205
return diopiSuccess;
210206
}
211207

0 commit comments

Comments
 (0)
Please sign in to comment.