Searched refs:initializeCudaContext (Results 1 – 4 of 4) sorted by relevance
213 TORCH_CUDA_CPP_API void initializeCudaContext();
924 void initializeCudaContext() { in initializeCudaContext() function1473 initializeCudaContext(); in jit_pwise_function()1643 initializeCudaContext(); in launch_jitted_pwise_function()
103 at::cuda::jit::initializeCudaContext(); in FusedKernelCUDA()
1096 at::cuda::jit::initializeCudaContext(); in call_with_numel()1216 at::cuda::jit::initializeCudaContext(); in call_raw()1265 at::cuda::jit::initializeCudaContext(); in CompileToNVRTC()