File tree 1 file changed +7
-11
lines changed
dipu/torch_dipu/csrc_dipu/diopirt
1 file changed +7
-11
lines changed Original file line number Diff line number Diff line change @@ -187,11 +187,9 @@ DIOPI_RT_API diopiError_t diopiGeneratorGetSeedAndOffset(
187
187
diopiGeneratorHandle_t th, uint64_t & seed, uint64_t & offset) {
188
188
auto generator = reinterpret_cast <at::Generator*>(th);
189
189
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 ();
195
193
196
194
return diopiSuccess;
197
195
}
@@ -200,12 +198,10 @@ DIOPI_RT_API diopiError_t diopiGeneratorSetSeedAndOffset(
200
198
diopiGeneratorHandle_t th, uint64_t seed, uint64_t offset) {
201
199
auto generator = reinterpret_cast <at::Generator*>(th);
202
200
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
+
209
205
return diopiSuccess;
210
206
}
211
207
You can’t perform that action at this time.
0 commit comments