From ff0e5909421a473eb910eead430333f182129230 Mon Sep 17 00:00:00 2001 From: mengyuanli Date: Tue, 11 Nov 2025 10:46:56 +0800 Subject: [PATCH] reshape_and_cahce/transdata --- docs/map_from_buildin_to_custom.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/map_from_buildin_to_custom.md b/docs/map_from_buildin_to_custom.md index 6cf29fd..508cbc2 100644 --- a/docs/map_from_buildin_to_custom.md +++ b/docs/map_from_buildin_to_custom.md @@ -1,4 +1,6 @@ |build-in接口|ms_custom_ops接口|变化说明| |------------------------------------|-------------------------------------|-----------------------------------| +| ops.auto_generate.format_cast | [ms_custom_ops.trans_data](../ops/c_api/trans_data/trans_data.md) | 两者均进行ND和NZ的相互转换。format_cast依赖ms内置白名单;trans_data不使用白名单但有使用约束,详见trans_data文档。同一权重只能选用其中一种进行转换,建议网络中统一采用同一种算子,两者不兼容。 | | ops.auto_generate.mla | [ms_custom_ops.mla](../ops/c_api/mla/mla_doc.md) | 新增了input_format参数,用于指定输入参数的format | +| ops.auto_generate.reshape_and_cache| [ms_custom_ops.reshape_and_cache](../ops/c_api/reshape_and_cache/reshape_and_cache.md) | 新增cache_mode参数,用于指定Atlas 训练系列cache的format是ND还是NZ; 新增head_num,cache_mode为NZ的时候必须提供,辅助计算。| | ops.moe_init_routing_v2 | [ms_custom_ops.moe_init_routing_v2](../ops/c_api/moe_init_routing_v2/moe_init_routing_v2.md) | 接口一致,仅支持 Atlas 推理系列 | -- Gitee