Searched refs:cuda_context (Results 1 – 1 of 1) sorted by relevance
206 ScopedActivateContext::ScopedActivateContext(GpuContext* cuda_context) { in ScopedActivateContext() argument211 if (tls->id == cuda_context->id()) { in ScopedActivateContext()213 CHECK_EQ(CurrentContext(), cuda_context->context()); in ScopedActivateContext()215 DCHECK_EQ(CurrentContext(), cuda_context->context()); in ScopedActivateContext()220 << " to " << cuda_context->id(); in ScopedActivateContext()225 CHECK_EQ(CUDA_SUCCESS, cuCtxSetCurrent(cuda_context->context())); in ScopedActivateContext()226 tls->id = cuda_context->id(); in ScopedActivateContext()227 tls->context = cuda_context; in ScopedActivateContext()