• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright (c) Meta Platforms, Inc. and affiliates.
2# All rights reserved.
3#
4# This source code is licensed under the BSD-style license found in the
5# LICENSE file in the root directory of this source tree.
6
7reduce:
8  parameter_names_with_default_values:
9    DTYPE: float
10    STORAGE: texture3d
11    INIT_ACCUM: VEC4_T(0)
12    UPDATE_ACCUM: accum + new_val
13    POSTPROCESS: accum
14  generate_variant_forall:
15    DTYPE:
16      - VALUE: half
17      - VALUE: float
18  shader_variants:
19    - NAME: sum
20    - NAME: mean
21      POSTPROCESS: (accum / tin_sizes[reduce_dim])
22    - NAME: amax
23      INIT_ACCUM: first_val
24      UPDATE_ACCUM: max(accum, new_val)
25      POSTPROCESS: accum
26    - NAME: amin
27      INIT_ACCUM: first_val
28      UPDATE_ACCUM: min(accum, new_val)
29      POSTPROCESS: accum
30