Searched refs:cudaStreamSynchronize (Results 1 – 21 of 21) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ctcloss_gpu_kernel.h | 158 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(stream), "cudaStreamSynchronize fai… in LaunchFirstHalf()
170 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(stream), "cudaStreamSynchronize fai… in LaunchFirstHalf()
184 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(stream), "cudaStreamSynchronize fai… in LaunchFirstHalf()
216 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(stream), "cudaStreamSynchronize fai… in LaunchSecondHalf()
266 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(stream), "cudaStreamSynchronize fai… in MemManageForCus()
|
| D | instance_norm_grad_gpu_kernel.h | 85 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stre… in Launch()
|
| D | lstm_grad_data_gpu_kernel.h | 99 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stre… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/nb14/ |
| D | dihedral_14_lj_force_impl.cu | 97 cudaStreamSynchronize(stream); in Dihedral14LJForce()
104 cudaStreamSynchronize(stream); in Dihedral14LJForce()
|
| D | dihedral_14_lj_atom_energy_impl.cu | 94 cudaStreamSynchronize(stream); in Dihedral14LJAtomEnergy()
|
| D | dihedral_14_lj_force_with_direct_cf_impl.cu | 109 cudaStreamSynchronize(stream); in Dihedral14LJForceWithDirectCF()
|
| /third_party/mindspore/tests/ut/cpp/stub/runtime/ |
| D | cuda_runtime_api.cc | 30 cudaError_t cudaStreamSynchronize(cudaStream_t stream) { return cudaSuccess; } in cudaStreamSynchronize() function
|
| D | cuda_runtime_api.h | 43 cudaError_t cudaStreamSynchronize(cudaStream_t stream);
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
| D | gpu_ps_cache.cc | 62 …CHECK_CUDA_RET_WITH_RETURN_ERROR_NOTRACE(cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stre… in SynchronizeStream()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/ |
| D | gpu_convert_to_dynamic_shape_gpu_kernel.h | 56 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(cuda… in PostExecute()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/pme/ |
| D | pme_excluded_force_update_kernel.h | 66 cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| D | pme_reciprocal_force_update_kernel.h | 147 cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| D | pme_energy_update_kernel.h | 159 cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | unique_gpu_kernel.h | 72 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stre… in PostExecute()
|
| D | dynamic_range_gpu_kernel.h | 101 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stre… in PostExecute()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/data/ |
| D | dataset_iterator_kernel.cc | 164 …CHECK_CUDA_RET_WITH_EXCEPT(kernel_node_, cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stre… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/lj/ |
| D | lj_force_with_pme_direct_force_update_kernel.h | 84 cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| D | lj_force_with_virial_energy_update_kernel.h | 92 cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | cuda_driver.cc | 166 auto ret = cudaStreamSynchronize((cudaStream_t)stream); in SyncStream()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/sponge/simple_constrain/ |
| D | constrain_kernel.h | 102 cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream_ptr)); in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/neighbor_list/ |
| D | neighbor_list_impl.cu | 454 cudaStreamSynchronize(stream); in NeighborListUpdate()
|