Home
last modified time | relevance | path

Searched refs:gpu_id (Results 1 – 25 of 47) sorted by relevance

12

/external/kernel-headers/original/uapi/linux/
Dkfd_ioctl.h52 __u32 gpu_id; /* to KFD */ member
86 __u32 gpu_id; /* to KFD */ member
105 __u32 gpu_id; /* to KFD */ member
118 __u32 gpu_id; /* from KFD */ member
136 __u32 gpu_id; /* to KFD */ member
141 __u32 gpu_id; /* to KFD */ member
147 __u32 gpu_id; /* to KFD */ member
153 __u32 gpu_id; /* to KFD */ member
211 __u32 gpu_id; member
237 __u32 gpu_id; /* to KFD */ member
[all …]
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_screen.h63 uint32_t gpu_id; /* 220, 305, etc */ member
108 return (screen->gpu_id >= 300) && (screen->gpu_id < 400); in is_a3xx()
114 return (screen->gpu_id >= 400) && (screen->gpu_id < 500); in is_a4xx()
120 return (screen->gpu_id >= 500) && (screen->gpu_id < 600); in is_a5xx()
Dfreedreno_screen.c636 screen->gpu_id = val; in fd_screen_create()
641 unsigned core = screen->gpu_id / 100; in fd_screen_create()
642 unsigned major = (screen->gpu_id % 100) / 10; in fd_screen_create()
643 unsigned minor = screen->gpu_id % 10; in fd_screen_create()
651 DBG(" GPU-id: %d", screen->gpu_id); in fd_screen_create()
666 switch (screen->gpu_id) { in fd_screen_create()
684 debug_printf("unsupported GPU: a%03d\n", screen->gpu_id); in fd_screen_create()
688 if (screen->gpu_id >= 500) { in fd_screen_create()
701 if ((screen->gpu_id >= 300) && (fd_device_version(dev) >= FD_VERSION_UNLIMITED_CMDS)) in fd_screen_create()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc99 int gpu_id; in ExtractGpuWithStreamAll() local
100 CHECK(strings::safe_strto32(ordered_capture, &gpu_id)); in ExtractGpuWithStreamAll()
101 return gpu_id; in ExtractGpuWithStreamAll()
128 int gpu_id; in ExtractGpuWithoutStream() local
129 CHECK(strings::safe_strto32(ordered_capture, &gpu_id)); in ExtractGpuWithoutStream()
130 return gpu_id; in ExtractGpuWithoutStream()
160 const int gpu_id = ExtractGpuWithStreamAll(device_name); in BuildCostModel() local
161 if (gpu_id >= 0) { in BuildCostModel()
163 gpu_hardware_stats.emplace(gpu_id, &device_stats); in BuildCostModel()
173 const int gpu_id = ExtractGpuWithoutStream(device_name.ToString()); in BuildCostModel() local
[all …]
/external/mesa3d/src/gallium/drivers/freedreno/ir3/
Dir3_compiler.c33 struct ir3_compiler * ir3_compiler_create(struct fd_device *dev, uint32_t gpu_id) in ir3_compiler_create() argument
37 compiler->gpu_id = gpu_id; in ir3_compiler_create()
Dir3_compiler.h38 uint32_t gpu_id; member
43 struct ir3_compiler * ir3_compiler_create(struct fd_device *dev, uint32_t gpu_id);
Dir3_cmdline.c54 bin = ir3_shader_assemble(so, so->shader->compiler->gpu_id); in dump_info()
157 unsigned gpu_id = 320; in main() local
255 gpu_id = strtol(argv[n+1], NULL, 0); in main()
327 s.compiler = ir3_compiler_create(NULL, gpu_id); in main()
Dir3_shader.c102 void * ir3_shader_assemble(struct ir3_shader_variant *v, uint32_t gpu_id) in ir3_shader_assemble() argument
106 bin = ir3_assemble(v->ir, &v->info, gpu_id); in ir3_shader_assemble()
110 if (gpu_id >= 400) { in ir3_shader_assemble()
131 uint32_t gpu_id = compiler->gpu_id; in assemble_variant() local
134 bin = ir3_shader_assemble(v, gpu_id); in assemble_variant()
617 if (ctx->screen->gpu_id >= 500) in max_tf_vtx()
Dir3.c132 if (info->gpu_id >= 500) { in emit_cat0()
134 } else if (info->gpu_id >= 400) { in emit_cat0()
561 uint32_t gpu_id) in ir3_assemble() argument
565 info->gpu_id = gpu_id; in ir3_assemble()
582 if (gpu_id >= 400) { in ir3_assemble()
/external/libdrm/freedreno/msm/
Dmsm_pipe.c62 *value = msm_pipe->gpu_id; in msm_pipe_get_param()
148 msm_pipe->gpu_id = get_param(pipe, MSM_PARAM_GPU_ID); in msm_pipe_new()
152 if (! msm_pipe->gpu_id) in msm_pipe_new()
156 INFO_MSG(" GPU-id: %d", msm_pipe->gpu_id); in msm_pipe_new()
Dmsm_priv.h56 uint32_t gpu_id; member
/external/tensorflow/tensorflow/core/grappler/
Ddevices.cc56 int64 AvailableGPUMemory(int gpu_id) { in AvailableGPUMemory() argument
60 CHECK_LT(gpu_id, gpu_platform->VisibleDeviceCount()); in AvailableGPUMemory()
62 gpu_platform->ExecutorForDevice(gpu_id).ValueOrDie(); in AvailableGPUMemory()
Ddevices.h34 int64 AvailableGPUMemory(int gpu_id);
/external/tensorflow/tensorflow/contrib/nccl/kernels/
Dnccl_ops.cc107 compute_stream->parent(), gpu_info->gpu_id, gpu_info->event_mgr, in ComputeAsync()
132 compute_stream->parent(), gpu_info->gpu_id, gpu_info->event_mgr, in ComputeAsync()
161 compute_stream->parent(), gpu_info->gpu_id, gpu_info->event_mgr, in ComputeAsync()
189 gpu_info->gpu_id, gpu_info->event_mgr, compute_stream, &c->input(0), in ComputeAsync()
221 gpu_info->gpu_id, gpu_info->event_mgr, compute_stream, out_t, in ComputeAsync()
/external/tensorflow/tensorflow/python/keras/_impl/keras/utils/
Dtraining_utils.py175 for i, gpu_id in enumerate(target_gpu_ids):
176 with ops.device('/gpu:%d' % gpu_id):
177 with ops.name_scope('replica_%d' % gpu_id):
/external/tensorflow/tensorflow/core/grappler/clusters/
Dutils.cc69 DeviceProperties GetLocalGPUInfo(int gpu_id) { in GetLocalGPUInfo() argument
75 cudaError_t error = cudaGetDeviceProperties(&properties, gpu_id); in GetLocalGPUInfo()
Dutils.h30 DeviceProperties GetLocalGPUInfo(int gpu_id);
Dsingle_machine.cc83 int gpu_id = 0; in Provision() local
89 attr = GetLocalGPUInfo(gpu_id++); in Provision()
/external/libdrm/freedreno/kgsl/
Dkgsl_pipe.c45 *value = kgsl_pipe->devinfo.gpu_id; in kgsl_pipe_get_param()
258 if (kgsl_pipe->devinfo.gpu_id >= 500) { in kgsl_pipe_new()
271 INFO_MSG(" GPU-id: %d", kgsl_pipe->devinfo.gpu_id); in kgsl_pipe_new()
/external/libdrm/freedreno/
Dfreedreno_ringbuffer.c118 assert(ring->pipe->gpu_id < 500); in fd_ringbuffer_reloc()
134 assert(ring->pipe->gpu_id < 500); in fd_ringbuffer_emit_reloc_ring()
Dfreedreno_pipe.c57 pipe->gpu_id = val; in fd_pipe_new()
/external/mesa3d/src/gallium/drivers/freedreno/a5xx/
Dfd5_screen.c101 screen->compiler = ir3_compiler_create(screen->dev, screen->gpu_id); in fd5_screen_init()
/external/mesa3d/src/gallium/drivers/freedreno/a4xx/
Dfd4_screen.c103 screen->compiler = ir3_compiler_create(screen->dev, screen->gpu_id); in fd4_screen_init()
/external/mesa3d/src/gallium/drivers/freedreno/a3xx/
Dfd3_screen.c104 screen->compiler = ir3_compiler_create(screen->dev, screen->gpu_id); in fd3_screen_init()
/external/mesa3d/src/gallium/drivers/freedreno/a2xx/
Dfd2_context.c115 (screen->gpu_id >= 220) ? a22x_primtypes : a20x_primtypes, in fd2_context_create()

12