Searched refs:gpu_device_id (Results 1 – 2 of 2) sorted by relevance
74 gpu_device_id(info->gpu_id), in Participant()104 const int gpu_device_id; member
274 if (a->gpu_device_id != b->gpu_device_id) { in GetCommunicator()275 return a->gpu_device_id < b->gpu_device_id; in GetCommunicator()387 devices[i] = collective->participants[i]->gpu_device_id; in GetCommunicator()643 p->gpu_device_id, " rank ", rank, " also marked as root."); in RunCollective()