Searched defs:deviceIdx (Results 1 – 6 of 6) sorted by relevance
649 auto deviceIdx = static_cast<c10::DeviceIndex>(rank % numGPUs); in getCUDADeviceForRank() local1189 int8_t deviceIdx = static_cast<int8_t>(c10::cuda::current_device()); in barrier() local
447 static void checkInput(const at::Tensor& input, int deviceIdx) { in checkInput()
423 size_t deviceIdx) { in newLikeFlat()
1302 … const uint32_t deviceIdx = context.getTestContext().getCommandLine().getVKDeviceId() - 1u; in queryDevGroupSurfacePresentCapabilitiesTest() local1411 … const uint32_t deviceIdx = context.getTestContext().getCommandLine().getVKDeviceId() - 1u; in queryDevGroupSurfacePresentModesTest() local
1632 const uint32_t deviceIdx = context.getTestContext().getCommandLine().getVKDeviceId() - 1u; in deviceGroupRenderTest() local1847 const uint32_t deviceIdx = context.getTestContext().getCommandLine().getVKDeviceId() - 1u; in deviceGroupRenderTest2() local
3331 …const uint32_t deviceIdx = vk::chooseDeviceIndex(context.getInstanceInterface(), instance, … in deviceGroupPeerMemoryFeatures() local