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 7addmm_optimized: 8 parameter_names_with_default_values: 9 DTYPE: float 10 MAT2_IS_TRANSPOSED: false 11 BATCH_MODE: false 12 TILE_ROWS: 4 13 HAS_BIAS: true 14 generate_variant_forall: 15 TILE_ROWS: 16 - VALUE: 4 17 SUFFIX: tile_row_4 18 - VALUE: 2 19 SUFFIX: tile_row_2 20 DTYPE: 21 - VALUE: float 22 - VALUE: half 23 shader_variants: 24 - NAME: addmm_optimized 25 - NAME: matmul_optimized 26 HAS_BIAS: false 27 - NAME: linear_optimized 28 MAT2_IS_TRANSPOSED: true 29 - NAME: matmul_transposed_optimized 30 MAT2_IS_TRANSPOSED: true 31 HAS_BIAS: false 32 - NAME: batch_addmm_optimized 33 BATCH_MODE: true 34 - NAME: batch_matmul_optimized 35 BATCH_MODE: true 36 HAS_BIAS: false 37 - NAME: batch_linear_optimized 38 MAT2_IS_TRANSPOSED: true 39 BATCH_MODE: true 40 - NAME: batch_matmul_transposed_optimized 41 MAT2_IS_TRANSPOSED: true 42 BATCH_MODE: true 43 HAS_BIAS: false 44