在 CUDA 10.0 中,以下 Runtime API 会导致 Context 切换:

  1. 'cudaSetDevice()'
  2. 'cudaDeviceSynchronize()'
  3. 'cudaStreamSynchronize()'
  4. 'cudaEventSynchronize()'
  5. 'cudaDeviceReset()'

当调用这些 API 时,当前的 Context 会被切换到指定的 Device 的 Context,或者等待当前 Device 的所有操作完成后再切换 Context。因此,在使用这些 API 时,需要注意 Context 切换的开销和影响。

CUDA 10.0 Runtime API 导致 Context 切换的函数

原文地址: https://www.cveoy.top/t/topic/n0AP 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录