在CUDA 10.0中,以下runtime API会导致context切换:

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

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

cuda100中哪些runtime api会导致context切换比如cudaSetDevice

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

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