Searched defs:gpu_id (Results 1 – 2 of 2) sorted by relevance
66 alloc_gpu_memory(uint16_t gpu_id) in alloc_gpu_memory()136 register_cpu_memory(uint16_t gpu_id) in register_cpu_memory()189 gpu_mem_cpu_map(uint16_t gpu_id) in gpu_mem_cpu_map()251 create_update_comm_flag(uint16_t gpu_id) in create_update_comm_flag()339 create_update_comm_list(uint16_t gpu_id) in create_update_comm_list()429 int16_t gpu_id = 0; in main() local
1410 uint8_t gpu_id; in cuda_gpu_remove() local