Searched refs:cuda_context (Results 1 – 1 of 1) sorted by relevance
199 ScopedActivateContext::ScopedActivateContext(GpuContext* cuda_context) { in ScopedActivateContext() argument208 VLOG(3) << "ScopedActivateContext switching to " << cuda_context->id(); in ScopedActivateContext()209 FAIL_IF_CUDA_RES_ERROR(cuCtxSetCurrent(cuda_context->context()), in ScopedActivateContext()212 tls->id = cuda_context->id(); in ScopedActivateContext()213 tls->context = cuda_context; in ScopedActivateContext()219 if (tls->id == cuda_context->id()) { in ScopedActivateContext()221 CHECK_EQ(CurrentContext(), cuda_context->context()); in ScopedActivateContext()223 DCHECK_EQ(CurrentContext(), cuda_context->context()); in ScopedActivateContext()228 << " to " << cuda_context->id(); in ScopedActivateContext()232 FAIL_IF_CUDA_RES_ERROR(cuCtxSetCurrent(cuda_context->context()), in ScopedActivateContext()[all …]