modified: ge/CMakeLists.txt

modified: ge/ge_inference.mk modified: ge/ge_runner.mk
5 years ago · 92fa54c0ed
parent 3413852064
commit 92fa54c0ed
3 changed files with 4 additions and 0 deletions
--- 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"
--- 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 \
--- 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 \