| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_lower_subgroups.c | 644 unsigned subgroup_size) in build_scan_reduce()
707 unsigned subgroup_size) in lower_scan_reduce()
840 nir_def *subgroup_size = nir_load_subgroup_size(b); in build_subgroup_mask() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
| H A D | lp_screen.c | 565 uint32_t *subgroup_size = ret; in llvmpipe_get_compute_param() local
571 uint32_t *subgroup_size = ret; in llvmpipe_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/core/ |
| H A D | kernel.rs | 325 subgroup_size: usize, field
1053 let subgroup_size = nir.subgroup_size(); in new() localVariable
1687 pub fn subgroup_size(&self) -> usize { in subgroup_size() method
1728 let subgroup_size = self.subgroup_size_for_block(dev, block); in subgroups_for_block() localVariable
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/ |
| H A D | si_get.c | 1337 unsigned subgroup_size; in si_get_compute_param() local
1350 uint32_t *subgroup_size = ret; in si_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/mesa/compiler/ |
| H A D | nir.rs | 319 pub fn subgroup_size(&self) -> u8 { in subgroup_size() method
320 let subgroup_size = unsafe { (*self.nir.as_ptr()).info.subgroup_size }; in subgroup_size() localVariable
|
| /aosp_15_r20/external/executorch/backends/vulkan/tools/gpuinfo/include/ |
| H A D | app.h | 29 uint32_t subgroup_size; variable
|
| /aosp_15_r20/external/mesa3d/src/intel/compiler/ |
| H A D | brw_nir_lower_cooperative_matrix.c | 60 unsigned subgroup_size; member
796 brw_nir_lower_cmat(nir_shader *shader, unsigned subgroup_size) in brw_nir_lower_cmat()
|
| /aosp_15_r20/external/mesa3d/src/vulkan/runtime/ |
| H A D | vk_nir.c | 121 enum gl_subgroup_size subgroup_size, in vk_spirv_to_nir()
|
| H A D | vk_shader.c | 119 enum gl_subgroup_size subgroup_size = vk_get_subgroup_size( in vk_shader_to_nir() local
|
| /aosp_15_r20/external/OpenCL-CTS/test_conformance/subgroups/ |
| H A D | subhelpers.cpp | 200 int subgroup_size, int &workgroup_size, in set_last_workgroup_params()
|
| H A D | subhelpers.h | 65 size_t subgroup_size; member
|
| /aosp_15_r20/external/mesa3d/src/asahi/compiler/ |
| H A D | agx_nir_lower_subgroups.c | 111 nir_def *subgroup_size = nir_imm_int(b, 32); in lower() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| H A D | mean.cc | 101 const uint32_t subgroup_size = ctx.gpu_info->vulkan_info.subgroup_size; in GenerateSubgroupBasedMean() local
|
| /aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/gpu/ |
| H A D | morton_radix_sort.h | 365 const uint subgroup_size = get_sub_group_size(); in sort_morton_codes_scatter_items() local
458 const uint subgroup_size = get_sub_group_size(); in sort_morton_codes_scatter_items() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/spirv/ |
| H A D | nir_spirv.h | 72 enum gl_subgroup_size subgroup_size; member
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/clover/core/ |
| H A D | device.cpp | 437 device::subgroup_size() const { in subgroup_size() function in device
|
| /aosp_15_r20/external/mesa3d/src/amd/vulkan/ |
| H A D | radv_shader_object.c | 97 const VkShaderRequiredSubgroupSizeCreateInfoEXT *const subgroup_size = in radv_shader_stage_init() local
|
| H A D | radv_pipeline.c | 154 const VkPipelineShaderStageRequiredSubgroupSizeCreateInfo *const subgroup_size = in radv_pipeline_get_shader_key() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | gpu_info.cc | 823 for (auto subgroup_size : supported_subgroup_sizes) { in SupportsSubGroupWithSize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_sharding_util.cc | 1677 int64_t subgroup_size = sharding.subgroup_types().size(); in GroupShardingOnDims() local
1705 int64_t subgroup_size = sharding.subgroup_types().size(); in GetManualSubgroupSharding() local
|
| H A D | hlo_verifier.cc | 369 int64_t subgroup_size = GetSubgroupSize(ag, group_mode); in CheckCommonAllGatherInvariants() local
448 int64_t subgroup_size = GetSubgroupSize(ars, group_mode); in HandleReduceScatter() local
|
| /aosp_15_r20/external/mesa3d/src/freedreno/ir3/ |
| H A D | ir3_nir.c | 712 unsigned subgroup_size = 0, max_subgroup_size = 0; in ir3_nir_post_finalize() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/r600/ |
| H A D | r600_pipe_common.c | 1020 uint32_t *subgroup_size = ret; in r600_get_compute_param() local
|
| /aosp_15_r20/external/mesa3d/src/microsoft/compiler/ |
| H A D | dxil_nir.c | 1984 nir_def *subgroup_size = nir_load_subgroup_size(b); in lower_num_subgroups() local
2173 nir_def *subgroup_size = nir_load_subgroup_size(b); in lower_subgroup_scan() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/ |
| H A D | shader_info.h | 160 enum gl_subgroup_size subgroup_size; member
|