CUDA 10.0 Runtime API 导致 Context 切换的函数
在 CUDA 10.0 中,以下 Runtime API 会导致 Context 切换:
- 'cudaSetDevice()'
- 'cudaDeviceSynchronize()'
- 'cudaStreamSynchronize()'
- 'cudaEventSynchronize()'
- 'cudaDeviceReset()'
当调用这些 API 时,当前的 Context 会被切换到指定的 Device 的 Context,或者等待当前 Device 的所有操作完成后再切换 Context。因此,在使用这些 API 时,需要注意 Context 切换的开销和影响。
原文地址: https://www.cveoy.top/t/topic/n0AP 著作权归作者所有。请勿转载和采集!