Home
last modified time | relevance | path

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

123

/external/kernel-headers/original/uapi/linux/
Dkfd_ioctl.h52 __u32 gpu_id; /* to KFD */ member
101 __u32 gpu_id; /* to KFD */ member
120 __u32 gpu_id; /* to KFD */ member
131 __u32 gpu_id; /* from KFD */ member
168 __u32 gpu_id; /* to KFD */ member
173 __u32 gpu_id; /* to KFD */ member
179 __u32 gpu_id; /* to KFD */ member
185 __u32 gpu_id; /* to KFD */ member
252 __u32 gpu_id; member
261 __u32 gpu_id; member
[all …]
/external/mesa3d/src/gallium/drivers/freedreno/
Dfreedreno_screen.h64 uint32_t gpu_id; /* 220, 305, etc */ member
119 return (screen->gpu_id >= 300) && (screen->gpu_id < 400); in is_a3xx()
125 return (screen->gpu_id >= 400) && (screen->gpu_id < 500); in is_a4xx()
131 return (screen->gpu_id >= 500) && (screen->gpu_id < 600); in is_a5xx()
Dfreedreno_screen.c830 screen->gpu_id = val; in fd_screen_create()
835 unsigned core = screen->gpu_id / 100; in fd_screen_create()
836 unsigned major = (screen->gpu_id % 100) / 10; in fd_screen_create()
837 unsigned minor = screen->gpu_id % 10; in fd_screen_create()
853 DBG(" GPU-id: %d", screen->gpu_id); in fd_screen_create()
868 switch (screen->gpu_id) { in fd_screen_create()
886 debug_printf("unsupported GPU: a%03d\n", screen->gpu_id); in fd_screen_create()
890 if (screen->gpu_id >= 500) { in fd_screen_create()
905 if ((screen->gpu_id >= 300) && (fd_device_version(dev) >= FD_VERSION_UNLIMITED_CMDS)) in fd_screen_create()
/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.c55 bin = ir3_shader_assemble(so, so->shader->compiler->gpu_id); in dump_info()
224 unsigned gpu_id = 320; in main() local
322 gpu_id = strtol(argv[n+1], NULL, 0); in main()
368 compiler = ir3_compiler_create(NULL, gpu_id); in main()
Dir3_shader.c108 void * ir3_shader_assemble(struct ir3_shader_variant *v, uint32_t gpu_id) in ir3_shader_assemble() argument
112 bin = ir3_assemble(v->ir, &v->info, gpu_id); in ir3_shader_assemble()
116 if (gpu_id >= 400) { in ir3_shader_assemble()
137 uint32_t gpu_id = compiler->gpu_id; in assemble_variant() local
140 bin = ir3_shader_assemble(v, gpu_id); in assemble_variant()
724 if (ctx->screen->gpu_id >= 500) in max_tf_vtx()
Dir3_nir.c133 if (shader->compiler->gpu_id >= 400) { in ir3_optimize_nir()
172 if (shader->compiler->gpu_id < 500) in ir3_optimize_nir()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc226 int gpu_id; in ExtractGpuWithStreamAll() local
227 CHECK(strings::safe_strto32(ordered_capture, &gpu_id)); in ExtractGpuWithStreamAll()
228 return gpu_id; in ExtractGpuWithStreamAll()
255 int gpu_id; in ExtractGpuWithoutStream() local
256 CHECK(strings::safe_strto32(ordered_capture, &gpu_id)); in ExtractGpuWithoutStream()
257 return gpu_id; in ExtractGpuWithoutStream()
287 const int gpu_id = ExtractGpuWithStreamAll(device_name); in BuildCostModel() local
288 if (gpu_id >= 0) { in BuildCostModel()
290 gpu_hardware_stats.emplace(gpu_id, &device_stats); in BuildCostModel()
300 const int gpu_id = ExtractGpuWithoutStream(string(device_name)); in BuildCostModel() local
[all …]
/external/perfetto/src/traced/probes/ftrace/test/data/synthetic/events/power/gpu_frequency/
Dformat10 field:unsigned int gpu_id; offset:12; size:4; signed:0;
12 print fmt: "state=%lu gpu_id=%lu", REC->state, REC->gpu_id
/external/tensorflow/tensorflow/core/grappler/
Ddevices.cc60 int64 AvailableGPUMemory(int gpu_id) { in AvailableGPUMemory() argument
64 CHECK_LT(gpu_id, gpu_platform->VisibleDeviceCount()); in AvailableGPUMemory()
65 se::StreamExecutor* se = gpu_platform->ExecutorForDevice(gpu_id).ValueOrDie(); in AvailableGPUMemory()
Ddevices.h34 int64 AvailableGPUMemory(int gpu_id);
/external/libdrm/freedreno/msm/
Dmsm_pipe.c58 *value = msm_pipe->gpu_id; in msm_pipe_get_param()
186 msm_pipe->gpu_id = get_param(pipe, MSM_PARAM_GPU_ID); in msm_pipe_new()
190 if (! msm_pipe->gpu_id) in msm_pipe_new()
194 INFO_MSG(" GPU-id: %d", msm_pipe->gpu_id); in msm_pipe_new()
/external/tensorflow/tensorflow/core/kernels/
Dnccl_ops.cc108 gpu_info->gpu_id, input, output, /*global_rank=*/-1, in ComputeAsync()
140 gpu_info->gpu_id, &c->input(0), /*output=*/nullptr, /*global_rank=*/-1, in ComputeAsync()
177 gpu_info->gpu_id, input, output, /*global_rank=*/-1, in ComputeAsync()
212 gpu_info->gpu_id, &c->input(0), /*output=*/nullptr, /*global_rank=*/-1, in ComputeAsync()
249 gpu_info->gpu_id, /*input=*/nullptr, output, /*global_rank=*/-1, in ComputeAsync()
/external/tensorflow/tensorflow/python/keras/utils/
Dmulti_gpu_utils.py220 for i, gpu_id in enumerate(target_gpu_ids):
221 with ops.device('/gpu:%d' % gpu_id):
222 with ops.name_scope('replica_%d' % gpu_id):
/external/tensorflow/tensorflow/core/grappler/clusters/
DBUILD30 "//tensorflow/core:gpu_id",
46 "//tensorflow/core:gpu_id",
121 "//tensorflow/core:gpu_id",
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator.h41 PlatformGpuId gpu_id, bool use_unified_memory, in GPUMemAllocator() argument
46 gpu_id_(gpu_id), in GPUMemAllocator()
Dgpu_device_test.cc34 int64 GetTotalGPUMemory(PlatformGpuId gpu_id) { in GetTotalGPUMemory() argument
36 GpuIdUtil::ExecutorForPlatformGpuId(GPUMachineManager(), gpu_id) in GetTotalGPUMemory()
44 Status GetComputeCapability(PlatformGpuId gpu_id, int* cc_major, in GetComputeCapability() argument
47 GpuIdUtil::ExecutorForPlatformGpuId(GPUMachineManager(), gpu_id) in GetComputeCapability()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc273 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
342 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
385 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
440 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
476 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
511 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
540 device->executor(), stream, event_mgr, device->gpu_id(), in TYPED_TEST()
/external/libdrm/freedreno/kgsl/
Dkgsl_pipe.c41 *value = kgsl_pipe->devinfo.gpu_id; in kgsl_pipe_get_param()
255 if (kgsl_pipe->devinfo.gpu_id >= 500) { in kgsl_pipe_new()
268 INFO_MSG(" GPU-id: %d", kgsl_pipe->devinfo.gpu_id); in kgsl_pipe_new()
/external/tensorflow/tensorflow/python/distribute/cluster_resolver/
Dslurm_cluster_resolver.py159 for gpu_id in range(gpu_offset, gpu_offset + self._gpus_per_task):
160 gpu_id_list.append(str(gpu_id))
/external/libdrm/freedreno/
Dfreedreno_ringbuffer.c114 assert(ring->pipe->gpu_id < 500); in fd_ringbuffer_reloc()
130 assert(ring->pipe->gpu_id < 500); in fd_ringbuffer_emit_reloc_ring()
Dfreedreno_pipe.c62 pipe->gpu_id = val; in fd_pipe_new2()
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_core_cpu.cmake69 "${tensorflow_source_dir}/tensorflow/core/common_runtime/gpu/gpu_id.h"
70 "${tensorflow_source_dir}/tensorflow/core/common_runtime/gpu/gpu_id.cc"
/external/perfetto/protos/perfetto/trace/ftrace/
Dpower.proto43 optional uint32 gpu_id = 1; field

123