Searched refs:compute_stream_ (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_bucket.cc | 103 MS_EXCEPTION_IF_NULL(compute_stream_); in CopyTensorToContiguousMemory()
106 cudaMemsetAsync(ar_input_addr_, 0, total_size_, static_cast<cudaStream_t>(compute_stream_)), in CopyTensorToContiguousMemory()
114 … memcpy_output_addrs_[i]->size, compute_stream_)) { in CopyTensorToContiguousMemory()
173 compute_stream_ = compute_streams.front(); in Init()
178 MS_EXCEPTION_IF_NULL(compute_stream_); in Init()
183 pre_event_->set_record_stream(compute_stream_); in Init()
186 post_event_->set_wait_stream(compute_stream_); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_bucket.cc | 116 … memcpy_input_addrs_[i]->size, RT_MEMCPY_DEVICE_TO_DEVICE, compute_stream_); in CopyTensorToContiguousMemory()
183 std::make_shared<AscendLaunchAtomicClean>(compute_stream_, tensor_type_list_[0], total_size_); in CreateLaunchAtomicClean()
193 compute_stream_ = compute_streams.front(); in Init()
198 MS_EXCEPTION_IF_NULL(compute_stream_); in Init()
204 pre_event_->set_record_stream(compute_stream_); in Init()
205 post_event_->set_wait_stream(compute_stream_); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | bucket.h | 38 compute_stream_(nullptr), in Bucket()
60 void *compute_stream_; variable
|