Searched defs:prior_device (Results 1 – 3 of 3) sorted by relevance
107 const auto prior_device = at::cuda::current_device(); in FusedKernelCUDA() local198 const auto prior_device = at::cuda::current_device(); in launch_raw() local
1090 const auto prior_device = at::cuda::current_device(); in call_with_numel() local1210 auto prior_device = at::cuda::current_device(); in call_raw() local1268 auto prior_device = at::cuda::current_device(); in CompileToNVRTC() local
499 const auto prior_device = prior_accumulator->input_metadata(0).device(); in set_data() local