.. Copyright (c) 2025 Huawei Technologies Co., Ltd. .. This program is free software, you can redistribute it and/or modify it under the terms and conditions of .. CANN Open Software License Agreement Version 2.0 (the "License"). .. Please refer to the License for details. You may not use this file except in compliance with the License. .. THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, .. INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE. .. See LICENSE in the root of the software repository for the full text of the License.

asc.lib.host

.. currentmodule:: asc.lib.host

Ascend C提供一组Matmul Tiling API,方便用户获取Matmul kernel计算时所需的Tiling参数。 用户只需要传入A/B/C矩阵的Position位置、Format格式和DType数据类型等信息,调用API接口,即可获取到Init中TCubeTiling结构体中的相关参数。

MatmulApiTiling/MultiCoreMatmulTiling/BatchMatmulTiling共有接口列表

.. autosummary:: :toctree: generated

MatmulApiTiling.enable_bias
MatmulApiTiling.get_base_k
MatmulApiTiling.get_base_m
MatmulApiTiling.get_base_n
MatmulApiTiling.get_tiling
MatmulApiTiling.set_a_layout
MatmulApiTiling.set_a_type
MatmulApiTiling.set_b_layout
MatmulApiTiling.set_b_type
MatmulApiTiling.set_batch_info_for_normal
MatmulApiTiling.set_batch_num
MatmulApiTiling.set_bias_type
MatmulApiTiling.set_buffer_space
MatmulApiTiling.set_c_layout
MatmulApiTiling.set_c_type
MatmulApiTiling.set_dequant_type
MatmulApiTiling.set_double_buffer
MatmulApiTiling.set_fix_split
MatmulApiTiling.set_mad_type
MatmulApiTiling.set_matmul_config_params
MatmulApiTiling.set_org_shape
MatmulApiTiling.set_shape
MatmulApiTiling.set_sparse
MatmulApiTiling.set_split_range
MatmulApiTiling.set_traverse

MultiCoreMatmulTiling其他接口

.. autosummary:: :toctree: generated

MultiCoreMatmulTiling.enable_multi_core_split_k
MultiCoreMatmulTiling.get_core_num
MultiCoreMatmulTiling.get_single_shape
MultiCoreMatmulTiling.set_align_split
MultiCoreMatmulTiling.set_dim
MultiCoreMatmulTiling.set_single_range
MultiCoreMatmulTiling.set_single_shape

BatchMatmulTiling其他接口

.. autosummary:: :toctree: generated

BatchMatmulTiling.get_core_num