From 53d85e3722305e823fdd62741b058d00e59b8687 Mon Sep 17 00:00:00 2001 From: PengC Date: Sat, 23 Nov 2024 16:37:44 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9tf2.x=E6=97=B6=E7=9A=84add=5F?= =?UTF-8?q?framework?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- operator/ascendc/0_introduction/1_add_frameworklaunch/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/operator/ascendc/0_introduction/1_add_frameworklaunch/README.md b/operator/ascendc/0_introduction/1_add_frameworklaunch/README.md index 71a6ac955..1a1132e6a 100644 --- a/operator/ascendc/0_introduction/1_add_frameworklaunch/README.md +++ b/operator/ascendc/0_introduction/1_add_frameworklaunch/README.md @@ -104,6 +104,8 @@ CANN软件包中提供了工程创建工具msopgen,AddCustom算子工程可通 - ASCEND_INSTALL_PATH:CANN软件包安装路径 脚本运行成功后,会在当前目录下创建CustomOp目录,编译完成后,会在CustomOp/build_out中,生成自定义算子安装包custom_opp_\_\.run,例如“custom_opp_ubuntu_x86_64.run”。 + + 注意:若tensorflow版本是2.x,则还需修改CustomOp/framework/tf_plugin/tensorflow_add_custom_plugin.cc文件,将OriginOpType内容改为"AddV2"。 备注:如果要使用dump调试功能,需要移除op_host内的Atlas 训练系列产品、Atlas 200/500 A2 推理产品的配置项。 -- Gitee