Searched defs:gpu_num (Results 1 – 5 of 5) sorted by relevance
46 std::optional<DrmDevice> create_drm_device(int fd, int32_t gpu_num) in create_drm_device()90 for (int32_t gpu_num = 0; gpu_num < num_devices; gpu_num++) { in create_all() local106 std::optional<DrmDevice> DrmDevice::create(int32_t gpu_num) in create()
89 uint32_t gpu_num = 0; in main() local147 auto gpu_num = device.gpu_num; in main() local
50 int32_t gpu_num = -1; variable
157 int32_t gpu_num) in add_group()
81 local_weight, sharded_dim, gpu_num, spec, rank argument