Searched refs:max_wg_size (Results 1 – 3 of 3) sorted by relevance
103 const uint32_t max_wg_size = in GenerateSubgroupBasedMean() local106 const uint32_t max_number_of_subgroups = max_wg_size / subgroup_size; in GenerateSubgroupBasedMean()109 if (width * height <= max_wg_size && width <= max_wg_size_x && in GenerateSubgroupBasedMean()116 wg_size_x = std::min({static_cast<uint32_t>(std::sqrt(max_wg_size)), in GenerateSubgroupBasedMean()118 wg_size_y = std::min({max_wg_size / wg_size_x, max_wg_size_y, in GenerateSubgroupBasedMean()
95 int3 max_wg_size; in GetWorkGroupsAlignedToGrid() local96 max_wg_size.x = gpu_info.GetMaxWorkGroupSizeForX(); in GetWorkGroupsAlignedToGrid()97 max_wg_size.y = gpu_info.GetMaxWorkGroupSizeForY(); in GetWorkGroupsAlignedToGrid()98 max_wg_size.z = gpu_info.GetMaxWorkGroupSizeForZ(); in GetWorkGroupsAlignedToGrid()100 grid, max_wg_size, kernel_info.max_work_group_size, work_groups); in GetWorkGroupsAlignedToGrid()321 int3 GetFirstSuitableWorkGroup(const std::vector<int3>& wgs, int max_wg_size) { in GetFirstSuitableWorkGroup() argument324 if (wg_size <= max_wg_size) { in GetFirstSuitableWorkGroup()
60 int3 GetFirstSuitableWorkGroup(const std::vector<int3>& wgs, int max_wg_size);