Browse Source

!589 Feature:add reformat kernel to fix same_transdata_breadth_fusion can not work

From: @hugo1
Reviewed-by: @sheng-nan,@xchu42
Signed-off-by: @xchu42
tags/v1.2.0
mindspore-ci-bot Gitee 3 years ago
parent
commit
299d645c88
3 changed files with 4 additions and 0 deletions
  1. +2
    -0
      ge/CMakeLists.txt
  2. +1
    -0
      ge/ge_inference.mk
  3. +1
    -0
      ge/ge_runner.mk

+ 2
- 0
ge/CMakeLists.txt View File

@@ -201,6 +201,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"
@@ -487,6 +488,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"


+ 1
- 0
ge/ge_inference.mk View File

@@ -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 \


+ 1
- 0
ge/ge_runner.mk View File

@@ -170,6 +170,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 \


Loading…
Cancel
Save