From 92fa54c0edc66016f3d20bb67993b087b7647d56 Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Fri, 11 Dec 2020 15:35:41 +0800 Subject: [PATCH] modified: ge/CMakeLists.txt modified: ge/ge_inference.mk modified: ge/ge_runner.mk --- ge/CMakeLists.txt | 2 ++ ge/ge_inference.mk | 1 + ge/ge_runner.mk | 1 + 3 files changed, 4 insertions(+) diff --git a/ge/CMakeLists.txt b/ge/CMakeLists.txt index b037f4a4..7c8ea305 100755 --- a/ge/CMakeLists.txt +++ b/ge/CMakeLists.txt @@ -202,6 +202,7 @@ set(TRAIN_SRC_LIST "host_kernels/sub_kernel.cc" "host_kernels/transdata_kernel.cc" "host_kernels/unpack_kernel.cc" + "host_kernels/reformat_kernel.cc" "graph/passes/folding_pass.cc" "graph/passes/get_original_format_pass.cc" "graph/passes/guarantee_const_pass.cc" @@ -488,6 +489,7 @@ set(INFER_SRC_LIST "host_kernels/slice_d_kernel.cc" "host_kernels/dynamic_stitch_kernel.cc" "host_kernels/identity_kernel.cc" + "host_kernels/reformat_kernel.cc" "graph/passes/stop_gradient_pass.cc" "graph/passes/prevent_gradient_pass.cc" "graph/passes/identity_pass.cc" diff --git a/ge/ge_inference.mk b/ge/ge_inference.mk index fe76a612..3503576f 100755 --- a/ge/ge_inference.mk +++ b/ge/ge_inference.mk @@ -164,6 +164,7 @@ OMG_HOST_SRC_FILES := \ host_kernels/slice_d_kernel.cc \ host_kernels/dynamic_stitch_kernel.cc \ host_kernels/identity_kernel.cc \ + host_kernels/reformat_kernel.cc \ graph/passes/stop_gradient_pass.cc \ graph/passes/prevent_gradient_pass.cc \ graph/passes/identity_pass.cc \ diff --git a/ge/ge_runner.mk b/ge/ge_runner.mk index 58ad1266..2b9ceb5c 100644 --- a/ge/ge_runner.mk +++ b/ge/ge_runner.mk @@ -171,6 +171,7 @@ LIBGE_LOCAL_SRC_FILES := \ host_kernels/sub_kernel.cc \ host_kernels/transdata_kernel.cc \ host_kernels/unpack_kernel.cc \ + host_kernels/reformat_kernel.cc \ graph/passes/folding_pass.cc \ graph/passes/get_original_format_pass.cc \ graph/passes/guarantee_const_pass.cc \