delete cce stub & AUX_SOURCE_DIRECTORY for graph

CMakeLists.txt

@@ -108,7 +108,7 @@ else ()
             elseif(PLATFORM STREQUAL "inference")
                 find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR})
                 find_module(runtime libruntime.so ${ASCEND_ACL_DIR})
-            find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
+                find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
                 find_module(msprofiler_ext libmsprofiler.a ${ASCEND_ACL_DIR})
                 if(PRODUCT STREQUAL "flr3")
                 elseif(PRODUCT STREQUAL "flr1")
@@ -119,12 +119,12 @@ else ()
                     find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
                 endif()
             elseif(PLATFORM STREQUAL "all")
-            find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR})
-            find_module(runtime libruntime.so ${ASCEND_RUNTIME_DIR})
+                find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR})
+                find_module(runtime libruntime.so ${ASCEND_RUNTIME_DIR})
                 find_module(msprofiler_fwk_ext libmsprofiler_fwk.a ${ASCEND_RUNTIME_DIR})
-            find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
-            find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
-            find_module(msprofiler_ext libmsprofiler.a ${ASCEND_ACL_DIR})
+                find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR})
+                find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR})
+                find_module(msprofiler_ext libmsprofiler.a ${ASCEND_ACL_DIR})
             else()
                 message(STATUS "PLATFORM param is invalid, should be train or inference, you choose nothing!")
             endif()

tests/CMakeLists.txt

@@ -15,7 +15,7 @@
 
 project(tests CXX C)
 find_package(Threads)
-add_subdirectory(depends/cce)
+
 add_subdirectory(depends/slog)
 add_subdirectory(depends/mmpa)
 add_subdirectory(depends/runtime)

tests/depends/cce/CMakeLists.txt

@@ -1,98 +0,0 @@
-# Copyright 2019-2020 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
-
-#cmake_minimum_required(VERSION 2.8)
-
-project(STUB_CCE)
-
-set(CMAKE_CXX_STANDARD 11)
-
-include_directories(${GE_CODE_DIR}/inc)
-include_directories(${GE_CODE_DIR}/inc/framework)
-include_directories(${GE_CODE_DIR}/metadef/inc/graph)
-include_directories(${GE_CODE_DIR}/inc/external)
-include_directories(${GE_CODE_DIR}/metadef/inc/external)
-include_directories(${GE_CODE_DIR}/metadef/inc/external/graph)
-include_directories(${GE_CODE_DIR}/metadef)
-include_directories(${GE_CODE_DIR}/metadef/inc)
-include_directories(${GE_CODE_DIR}/metadef/graph)
-include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc)
-include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce)
-include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/ops)
-include_directories(${CMAKE_BINARY_DIR})
-include_directories(${CMAKE_BINARY_DIR}/proto/ge)
-set(PROTO_LIST
-    "${GE_CODE_DIR}/metadef/proto/om.proto"
-    "${GE_CODE_DIR}/metadef/proto/ge_ir.proto"
-    "${GE_CODE_DIR}/metadef/proto/task.proto"
-)
-
-protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST})
-
-set(SRCS
-    "${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
-    "${GE_CODE_DIR}/metadef/graph/anchor.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
-    "${GE_CODE_DIR}/metadef/graph/buffer.cc"
-    "${GE_CODE_DIR}/metadef/graph/aligned_ptr.cc"
-    "${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
-    "${GE_CODE_DIR}/metadef/graph/graph.cc"
-    "${GE_CODE_DIR}/metadef/graph/model.cc"
-    "${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
-    "${GE_CODE_DIR}/metadef/graph/node.cc"
-    "${GE_CODE_DIR}/metadef/graph/op_desc.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
-    "${GE_CODE_DIR}/metadef/graph/tensor.cc"
-    "${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/dumper/ge_graph_dumper.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
-    "${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
-    "${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
-    "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
-)
-add_library(cce_ge_stub SHARED src/cce_stub.cc ${PROTO_SRCS} ${PROTO_HDRS})
-
-target_compile_definitions(cce_ge_stub PRIVATE
-    google=ascend_private
-)
-
-target_link_libraries(cce_ge_stub
-    $<BUILD_INTERFACE:intf_pub>
-    -Wl,--no-as-needed
-    ascend_protobuf
-    -Wl,--as-needed
-    c_sec
-)
-
-add_library(cce_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS})
-
-target_compile_definitions(cce_stub PRIVATE
-    google=ascend_private
-)
-
-target_link_libraries(cce_stub PRIVATE
-    $<BUILD_INTERFACE:intf_pub>
-    -Wl,--no-as-needed
-    ascend_protobuf
-    -Wl,--as-needed
-    c_sec
-)

tests/depends/cce/src/cce_stub.cc

@@ -1,576 +0,0 @@
-/**
- * Copyright 2019-2020 Huawei Technologies Co., Ltd
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include <vector>
-#include <cce/cce.h>
-#include <cce/dnn.h>
-#include <cce/compiler_stub.h>
-#include <cce/taskdown_api.h>
-
-#include "cce/optimizer/fusion_engine.h"
-#include "common/op/attr_value_util.h"
-#include "graph/utils/tensor_utils.h"
-#include "graph/utils/graph_utils.h"
-
-using namespace cce;
-using namespace std;
-using namespace ge;
-using namespace fusion;
-
-uint64_t global_mem_base = 0;
-
-namespace cce {
-#define DIM_MAX_SIZE 8
-static const uint32_t C0 = 16;
-struct tagCcPad {};
-struct tagCcConvolution {};
-
-struct tagCcLRN {};
-
-struct tagCcFasterRcnnProposal {};
-struct tagCcRoiAlign {};
-struct tagCcBatchNorm {};
-struct tagCcDetectpostprocess {};
-
-struct tagCcSsdDetectionOutput {};
-
-struct tagCcRefinedetDetectionOutput {};
-
-struct tagCcMsrGenerateRpnProposals {};
-
-struct tagCcFilter {
-  vector<uint32_t> dims;
-};
-
-struct tagCcTensor {
-  ccTensorFormat_t format;
-  ccDataType_t data_type;
-  uint32_t dim_cnt;
-  int32_t real_dim_cnt;
-  uint32_t data_size;
-  int32_t dim_buf[DIM_MAX_SIZE];
-  int32_t stride_buf[DIM_MAX_SIZE];
-};
-
-typedef struct tagCcPooling {
-  ccPoolingMode_t mode;
-  ccPaddingMode_t pad_mode;
-  ccNanPropagation_t max_pooling_nan_opt;
-  uint32_t dim_cnt;
-  int32_t window_dim[6];
-  int32_t padding[6];
-  int32_t stride[6];
-} ccPooling_t;
-
-struct tagCcActivation {};
-
-struct tagCcFasterRcnnDetectionOutput {};
-struct tagCcSpatialTransformer {};
-
-struct tagCcPower {};
-struct tagCcResizeBilinear {};
-struct tagCcSsdNormalize {};
-struct tagCcSsdPostProcessor {};
-struct tagCcSsdPriorBox {};
-struct tagCcPsRoiPooling {};
-
-struct tagMsrFastRcnnPredictions {};
-struct tagCcPRelu {};
-struct tagCcStridedSlice {};
-
-struct tagCcStridedSliceAttrs {};
-
-struct tagCcRnn {};
-
-struct tagCcArgmaxmin {};
-
-typedef struct tagCcLog {
-  ccDataType_t data_type;
-  uint32_t param_cnt;
-} ccLog_t;
-typedef struct tagCcLog *ccLogDescriptor_t;
-
-struct tagCcPadV2 {};
-
-ccStatus_t ccGetPadV2OutputDim(const ccTensorDescriptor_t x_desc, const ccPadV2Descriptor_t pad_desc, int32_t *dim_cnt,
-                               int32_t dim[], int32_t dim_len) {
-  *dim_cnt = 4;
-  dim[0] = 1;
-  dim[1] = 2;
-  dim[2] = 2;
-  dim[3] = 3;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccPadV2Forward(ccHandle_t handle, const ccPadV2Descriptor_t pad_desc, const void *alpha,
-                          const ccTensorDescriptor_t x_desc, const void *x, const void *beta,
-                          const ccTensorDescriptor_t output_desc, void *output) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccCreatePadV2Descriptor(ccPadV2Descriptor_t *pad_desc) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccDestroyPadV2Descriptor(ccPadV2Descriptor_t *pad_desc) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccSetKernelOpMap(ccHandle_t handle) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccDataDumpForward(ccHandle_t handle, const void *buffer, const uint64_t buf_len, const uint32_t task_index) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetPadV2Descriptor(ccPadV2Descriptor_t pad_desc, const int32_t pad_shape_cnt,
-                                const int32_t pad_shape_low[], const int32_t pad_shape_high[],
-                                const ccPadMode_t pad_mode, const void *pad_value, const ccDataType_t pad_value_type) {
-  return CC_STATUS_SUCCESS;
-}
-
-struct tagCcYoloDetectionOutput {
-  ccYoloVersion_t yolo_version;
-  uint32_t net_h;
-  uint32_t net_w;
-  uint32_t post_top_k;
-  uint32_t classes;
-  float nms_threshold;
-  float iou_thre_decay;
-  float coor_scale_factor;
-  bool relative;
-  float obj_threshold;
-  float cls_threshold;
-  uint32_t bias_num;
-  float *bias;
-};
-
-struct tagCcYoloRegion {};
-
-struct tagCcEltwise {};
-
-struct tagCcHashTableLookup {};
-
-struct tagCcEmbeddingAttnDecoder {};
-struct tagNonMaxSuppression {};
-
-struct tagCcArcSinCos {};
-struct tagCcPow {};
-struct tagCcConcatFive2Four_t {};
-struct tagCcConcatFour2Five_t {};
-
-ccStatus_t ccCreatePowDescriptor(ccPowDescriptor_t *pow_desc) {
-  *pow_desc = new tagCcPow();
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetPowDescriptor(ccPowDescriptor_t pow_desc, ccDataType_t data_type, uint32_t param_cnt) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccDestroyPowDescriptor(ccPowDescriptor_t *pow_desc) {
-  if (nullptr == pow_desc) {
-    return CC_STATUS_BAD_PARAM;
-  }
-
-  delete *pow_desc;
-  *pow_desc = 0;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccPowForward(ccHandle_t handle, const ccPowDescriptor_t pow_desc, const void *pow_param, const void *alpha,
-                        const ccTensorDescriptor_t x_desc, const void *x, const ccTensorDescriptor_t y_desc,
-                        const void *y, const void *beta, const ccTensorDescriptor_t z_desc, void *z) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccLogicalOrForward(ccHandle_t handle, const void *alpha, const ccTensorDescriptor_t x_desc, const void *x,
-                              const ccTensorDescriptor_t y_desc, const void *y, const void *beta,
-                              const ccTensorDescriptor_t output_desc, void *output) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccCompareForward(ccHandle_t handle, ccCompareType_t compare_type, const void *alpha,
-                            const ccTensorDescriptor_t x_desc, const void *x, const ccTensorDescriptor_t y_desc,
-                            const void *y, const void *beta, const ccTensorDescriptor_t output_desc, void *output) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccGetCompareOutputDim(const ccTensorDescriptor_t x_desc, const ccTensorDescriptor_t y_desc, int32_t *dim_cnt,
-                                 int32_t *dim, int32_t dim_len) {
-  *dim_cnt = 4;
-  dim[0] = 1;
-  dim[1] = 1;
-  dim[2] = 1;
-  dim[3] = 1;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccArcTanForward(ccHandle_t handle, const void *alpha, const ccTensorDescriptor_t x_desc, const void *x,
-                           const void *beta, const ccTensorDescriptor_t y_desc, void *y) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccAtanhForward(ccHandle_t handle, const void *alpha, const ccTensorDescriptor_t x_desc, const void *x,
-                          const void *beta, const ccTensorDescriptor_t y_desc, void *y) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccIsDepthwiseHighPerformance(int32_t input_n, int32_t input_c, int32_t input_h, int32_t input_w,
-                                        int32_t filter_n, int32_t filter_c, int32_t filter_h, int32_t filter_w,
-                                        int32_t dilation_h, int32_t dilation_w, int32_t pad_h_head, int32_t pad_h_tail,
-                                        int32_t pad_w_head, int32_t pad_w_tail, int32_t stride_h, int32_t stride_w,
-                                        int32_t group_num, bool &is_high_performance, bool is_quant,
-                                        ccDataType_t input_data_type, ccDataType_t output_data_type) {
-  is_high_performance = true;
-  return CC_STATUS_SUCCESS;
-}
-
-struct tagCcSpaceToBatch {};
-
-struct tagCcBatchToSpace {};
-
-struct tagCcResizeNearestNeighbor {};
-
-ccStatus_t ccGetStream(ccHandle_t handle, rtStream_t *stream_id) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccGetRtVersion(uint32_t *count) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccDestroyTensorDescriptor(ccTensorDescriptor_t *tensor_desc) {
-  if (nullptr == tensor_desc) {
-    return CC_STATUS_BAD_PARAM;
-  }
-  delete *tensor_desc;
-  *tensor_desc = 0;
-  return CC_STATUS_SUCCESS;
-}
-ccStatus_t ccDestroyFilterDescriptor(ccFilterDescriptor_t *filter_desc) {
-  delete *filter_desc;
-  *filter_desc = 0;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccGetFilterSizeInBytes(const ccFilterDescriptor_t filter_desc, uint32_t *size) {
-  *size = filter_desc->dims[0] * filter_desc->dims[1] * filter_desc->dims[2] * filter_desc->dims[3] * sizeof(float);
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccTransFilter(const ccFilterDescriptor_t w_desc, const void *w, ccFilterDescriptor_t y_desc, void *y,
-                         uint32_t y_size_in_bytes) {
-  y = const_cast<void *>(w);
-
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccCreateTensorDescriptor(ccTensorDescriptor_t *tensor_desc) {
-  *tensor_desc = new tagCcTensor();
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetTensor4dDescriptor(ccTensorDescriptor_t tensor_desc, ccTensorFormat_t format, ccDataType_t data_type,
-                                   int32_t n, int32_t c, int32_t h, int32_t w) {
-  if (CC_TENSOR_NHWC == format) {
-    tensor_desc->dim_buf[0] = n;
-    tensor_desc->dim_buf[1] = h;
-    tensor_desc->dim_buf[2] = w;
-    tensor_desc->dim_buf[3] = c;
-  } else {
-    tensor_desc->dim_buf[0] = n;
-    tensor_desc->dim_buf[1] = c;
-    tensor_desc->dim_buf[2] = h;
-    tensor_desc->dim_buf[3] = w;
-  }
-  tensor_desc->dim_cnt = 4;
-  tensor_desc->data_type = data_type;
-  tensor_desc->format = format;
-  tensor_desc->data_size = n * c * h * w * sizeof(data_type);
-  return CC_STATUS_SUCCESS;
-}
-ccStatus_t ccGetTensorSizeInBytes(const ccTensorDescriptor_t tensor_desc, uint32_t *size) {
-  if ((NULL == tensor_desc) || (NULL == size)) {
-    return CC_STATUS_BAD_PARAM;
-  }
-  *size = tensor_desc->data_size;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccGetTensorMemorySizeInBytes(const ccTensorDescriptor_t tensor_desc, uint32_t *size) {
-  *size = tensor_desc->data_size;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccCreateFilterDescriptor(ccFilterDescriptor_t *filter_desc) {
-  *filter_desc = new tagCcFilter();
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetFilter4dDescriptor(ccFilterDescriptor_t filter_desc, ccTensorFormat_t format, ccDataType_t data_type,
-                                   int32_t k, int32_t c, int32_t h, int32_t w) {
-  filter_desc->dims.push_back(k);
-  filter_desc->dims.push_back(c);
-  filter_desc->dims.push_back(h);
-  filter_desc->dims.push_back(w);
-
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetFilterFractalDescriptor(ccFilterDescriptor_t filter_desc, ccTensorFormat_t format,
-                                        ccDataType_t data_type, int32_t k, int32_t c, int32_t h, int32_t w) {
-  filter_desc->dims.push_back(k);
-  filter_desc->dims.push_back(c);
-  filter_desc->dims.push_back(h);
-  filter_desc->dims.push_back(w);
-
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetStream(ccHandle_t handle, rtStream_t stream_id) { return CC_STATUS_SUCCESS; }
-ccStatus_t ccCreatePoolingMaskDescriptor(ccTensorDescriptor_t *pooling_mask_desc) {
-  *pooling_mask_desc = new tagCcTensor();
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetPoolingMaskTensorDescriptor(ccTensorDescriptor_t tensor_desc, ccTensorFormat_t format,
-                                            ccDataType_t data_type, int32_t n, int32_t c, int32_t h, int32_t w,
-                                            int32_t window_h, int32_t window_w) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetFilter6dDescriptor(ccTensorDescriptor_t filter_desc, ccTensorFormat_t format, ccDataType_t data_type,
-                                   int32_t c1, int32_t h, int32_t w, int32_t n, int32_t co, int32_t c0) {
-  return CC_STATUS_SUCCESS;
-}
-
-/// @ingroup dnn
-/// @brief get the format and dimcnt of GeTensor
-/// @param [in] tensor_desc   descriptor of tensor
-/// @param [in|out] format   point to format
-/// @return ccStatus_t
-ccStatus_t ccGetTensorFormat(const ccTensorDescriptor_t tensor_desc, ccTensorFormat_t *format) {
-  *format = tensor_desc->format;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccTransTensor(const ccTensorDescriptor_t x_desc, const void *x, const ccTensorDescriptor_t y_desc, void *y,
-                         uint32_t y_size_in_bytes) {
-  return CC_STATUS_SUCCESS;
-}
-void cceSysInit() {}
-
-bool compilerStubFree() { return true; }
-
-bool compilerStubInit() { return true; }
-
-ccStatus_t ccSetInt8Filter4dDescriptor(ccFilterDescriptor_t filter_desc, ccTensorFormat_t format,
-                                       ccDataType_t data_type, int32_t k, int32_t c, int32_t h, int32_t w,
-                                       ccDataType_t output_data_type) {
-  filter_desc->dims.push_back(k);
-  filter_desc->dims.push_back(c);
-  filter_desc->dims.push_back(h);
-  filter_desc->dims.push_back(w);
-
-  return CC_STATUS_SUCCESS;
-}
-ccStatus_t ccSetTensorNdDescriptor(ccTensorDescriptor_t tensor_desc, ccDataType_t data_type, int32_t dim_cnt,
-                                   int32_t dimA[]) {
-  tensor_desc->data_type = data_type;
-  tensor_desc->data_size = sizeof(data_type);
-  for (int32_t i = 0; i < dim_cnt; i++) {
-    tensor_desc->data_size = tensor_desc->data_size * dimA[i];
-  }
-  tensor_desc->format = CC_TENSOR_ND;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t CceProfilingConfig(const char *target, const char *job_ctx, uint32_t flag) { return CC_STATUS_SUCCESS; }
-ccStatus_t ccSetTensorRealDimCnt(ccTensorDescriptor_t tensor_desc, int32_t real_dim_cnt) {
-  if (tensor_desc != NULL && tensor_desc != nullptr) {
-    tensor_desc->real_dim_cnt = real_dim_cnt;
-  }
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccGetTensorRealDimCnt(ccTensorDescriptor_t tensor_desc, int32_t *real_dim_cnt) {
-  *real_dim_cnt = tensor_desc->real_dim_cnt;
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetQuantizeFactors(ccQuantizeDescriptor_t quantize_info, ccScaleValueMode_t scale_val_mode,
-                                const uint16_t *scale, const uint16_t *offset, const uint8_t *offset_pad) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetReQuantizeFactors(ccQuantizeDescriptor_t quantize_info, ccScaleValueMode_t scale_val_mode,
-                                  const uint16_t *scale_rq, const uint16_t *next_layer_offset,
-                                  const int32_t *offset_w) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetDeQuantizeFactors(ccQuantizeDescriptor_t quantize_info, ccScaleValueMode_t scale_val_mode,
-                                  const uint16_t *scale_dq, const int32_t *offset_w) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetQuantizeAlgoAndScaleType(ccQuantizeDescriptor_t quantize_info, ccQuantizeAlgo_t quant_algo,
-                                         ccScaleType_t scale_type, bool relu_flag) {
-  return CC_STATUS_SUCCESS;
-}
-ccStatus_t ccPrintTimeStat() { return CC_STATUS_SUCCESS; }
-ccStatus_t ccSetModelId(ccHandle_t handle, uint32_t model_id) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccGetKernelContext(rtStream_t stream_id, ccOpContext &op_context) {
-  if (stream_id == nullptr) {
-    op_context.kernelType = ccKernelType::TE;
-  } else {
-    op_context.kernelType = ccKernelType::CCE_AI_CORE;
-    op_context.opId = 1;
-    op_context.kernelFuncId = 1;
-    op_context.isFlowtable = true;
-    op_context.opCount = 1;
-    op_context.opIndex2[0] = 0;
-  }
-
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccUpdateKernelArgs(ccOpContext &op_context, uint64_t data_base_addr, uint64_t weight_base_addr,
-                              uint64_t variable_base_addr, void *args_addr, uint64_t args_size, void *l2ctrl_addr) {
-  return CC_STATUS_SUCCESS;
-}
-ccStatus_t ccGetKernelArgsAddrs(ccOpContext &op_context, void *args_addr, uint64_t args_size, void *l2ctrl_addr,
-                                std::vector<ccOpAddrsInfo> &op_addrs_info) {
-  // cce
-  ccOpAddrsInfo tmp_op_addrs_info;
-  uint64_t tmp_input = (uint64_t)global_mem_base;
-  tmp_op_addrs_info.addrPos = &tmp_input;
-  tmp_op_addrs_info.addrData = tmp_input;
-  op_addrs_info.push_back(tmp_op_addrs_info);
-
-  uint64_t tmp_output = (uint64_t)(global_mem_base + 5476352);
-  tmp_op_addrs_info.addrPos = &tmp_output;
-  tmp_op_addrs_info.addrData = tmp_output;
-  op_addrs_info.push_back(tmp_op_addrs_info);
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccSetKernelArgs(std::vector<ccOpAddrsInfo> &date_info) { return CC_STATUS_SUCCESS; }
-}  // namespace cce
-// ccFusion no namespace
-ccStatus_t ccFusionStart(ccHandle_t handle, uint32_t graph_id, uint32_t init_flag, CceFusionMemCfg_t mem_cfg) {
-  return CC_STATUS_SUCCESS;
-}
-
-//???ccFusion ????namespace cce??
-ccStatus_t ccFusionStart(ccHandle_t handle, uint32_t graph_id, uint32_t init_flag, uint32_t addr_change_flag) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t ccFusionEnd(ccHandle_t handle, uint32_t graph_id) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccFusionTaskEnd(ccHandle_t handle, uint32_t graph_id) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccKernelLaunchRepeat(ccHandle_t handle) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t ccKernelDelete(ccHandle_t handle) { return CC_STATUS_SUCCESS; }
-
-ccStatus_t cce::ccSetTensorFormat(cce::tagCcTensor *, cce::tagCcTensorFormat) { return CC_STATUS_SUCCESS; }
-
-namespace fusion {
-uint32_t BufferFusion(std::shared_ptr<ge::ComputeGraph>, std::shared_ptr<ge::ComputeGraph>, bool) { return 0; }
-
-uint32_t BufferFusionTrain(std::shared_ptr<ge::ComputeGraph>, std::shared_ptr<ge::ComputeGraph>) { return 0; }
-
-uint32_t GraphFusionTrain(ge::ComputeGraphPtr orig_graph, ge::ComputeGraphPtr fusion_graph) { return 0; }
-}  // namespace fusion
-namespace fusion {
-using namespace ge;
-
-uint32_t Fusion(ComputeGraphPtr model_graph, ComputeGraphPtr fusion_graph, kScopeNodeMap_t &te_fusion_map) {
-  OpDescPtr op_def_a = std::make_shared<OpDesc>();
-  op_def_a->SetName("reduction_nd");
-  op_def_a->SetType("reduction_nd");
-
-  GeTensorDescPtr v_input_desc = std::make_shared<GeTensorDesc>();
-  op_def_a->AddInputDesc(*v_input_desc);
-
-  vector<int64_t> v_input;
-  v_input.push_back(0);
-  op_def_a->SetInputOffset(v_input);
-
-  GeTensorDesc input_desc = op_def_a->GetInputDesc(0);
-  input_desc.SetFormat(FORMAT_NCHW);
-  input_desc.SetDataType(DT_FLOAT);
-  input_desc.SetShape(GeShape({1, 3, 5, 5}));
-  ge::TensorUtils::SetSize(input_desc, 192);
-  ge::TensorUtils::SetRealDimCnt(input_desc, 4);
-
-  GeTensorDescPtr output_desc = std::make_shared<GeTensorDesc>();
-  op_def_a->AddOutputDesc(*output_desc);
-
-  output_desc->SetFormat(FORMAT_NCHW);
-  output_desc->SetDataType(DT_FLOAT);
-  output_desc->SetShape(GeShape({1, 3, 5}));
-  ge::TensorUtils::SetSize(*output_desc, 96);
-  ge::TensorUtils::SetRealDimCnt(*output_desc, 3);
-
-  OpDescPtr op_def_b = std::make_shared<OpDesc>();
-  op_def_b->SetName("transdata_1");
-  op_def_b->SetType("TransData");
-
-  int stream_num = 1;
-  int flag = 0;
-
-  NodePtr node_a = fusion_graph->AddNode(op_def_a);
-  NodePtr node_b = fusion_graph->AddNode(op_def_b);
-
-  GraphUtils::AddEdge(node_a->GetOutDataAnchor(0), node_b->GetInDataAnchor(0));
-  int32_t a = 1;
-  int32_t b = 2;
-
-  AttrUtils::SetInt(op_def_a, "fusion_scope", a);
-  AttrUtils::SetInt(op_def_b, "fusion_scope", b);
-
-  vector<NodePtr> node_list1;
-  node_list1.push_back(node_a);
-  vector<NodePtr> node_list2;
-  node_list2.push_back(node_b);
-  te_fusion_map[1] = node_list1;
-  te_fusion_map[2] = node_list2;
-
-  return FUSION_STATUS_SUCCESS;
-}
-
-uint32_t FusionTaskBuild(cce::ccHandle_t cc_handle, ge::ComputeGraphPtr fusion_graph, ge::Buffer &buffer,
-                         ModelRes &model_res, std::vector<TaskDef> &task_def_list_) {
-  TaskDef task_def_temp;
-  task_def_list_.push_back(task_def_temp);
-
-  return FUSION_STATUS_SUCCESS;
-}
-uint32_t GraphFusion(ge::ComputeGraphPtr orig_graph, ge::ComputeGraphPtr fusion_graph) {
-  *fusion_graph = *orig_graph;
-  return FUSION_STATUS_SUCCESS;
-}
-
-void FusionTaskBuildComplete(std::vector<ccHandle_t> cc_handle_list) { return; }
-
-}  // namespace fusion
-
-ccStatus_t cce::ccSetTensorDescriptorQuantizeParam(ccTensorDescriptor_t tensor_desc,
-                                                   const ccVecQuantizePara_t *vec_quantize_para) {
-  return CC_STATUS_SUCCESS;
-}
-
-ccStatus_t cce::ccSetAllOffsetQuantizeFactors(ccQuantizeDescriptor_t quantize_info, const uint8_t *offset_w,
-                                              const uint8_t *offset_d, const uint16_t *scale_req,
-                                              const uint16_t *offset_d_next) {
-  return CC_STATUS_SUCCESS;
-}

tests/depends/cce/src/op_kernel_registry.cc

@@ -1,29 +0,0 @@
-#include "register/op_kernel_registry.h"
-
-namespace ge {
-class OpKernelRegistry::OpKernelRegistryImpl {
-
-};
-
-OpKernelRegistry::OpKernelRegistry() {
-}
-
-OpKernelRegistry::~OpKernelRegistry() {
-
-}
-
-bool OpKernelRegistry::IsRegistered(const std::string &op_type) {
-  return false;
-}
-
-std::unique_ptr<HostCpuOp> OpKernelRegistry::CreateHostCpuOp(const std::string &op_type) {
-  return nullptr;
-}
-
-void OpKernelRegistry::RegisterHostCpuOp(const std::string &op_type, CreateFn create_fn) {
-}
-
-HostCpuOpRegistrar::HostCpuOpRegistrar(const char *op_type, HostCpuOp *(*create_fn)()) {
-
-}
-} // namespace ge

tests/ut/common/graph/CMakeLists.txt

@@ -61,53 +61,26 @@ set(UT_FILES
     "testcase/ge_graph/ge_model_unittest.cc"
 )
 
-set(SRC_FILES
-    "${GE_CODE_DIR}/metadef/graph/option/ge_local_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/option/ge_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/anchor.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
-    "${GE_CODE_DIR}/metadef/graph/attr_value.cc"
-    "${GE_CODE_DIR}/metadef/graph/buffer.cc"
-    "${GE_CODE_DIR}/metadef/graph/aligned_ptr.cc"
-    "${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
-    "${GE_CODE_DIR}/metadef/graph/graph.cc"
-    "${GE_CODE_DIR}/metadef/graph/gnode.cc"
-    "${GE_CODE_DIR}/metadef/graph/ascend_string.cc"
-    "${GE_CODE_DIR}/metadef/graph/model.cc"
-    "${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
-    "${GE_CODE_DIR}/metadef/graph/node.cc"
-    "${GE_CODE_DIR}/metadef/graph/op_desc.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
-    "${GE_CODE_DIR}/metadef/graph/tensor.cc"
-    "${GE_CODE_DIR}/metadef/graph/types.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
-    "${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
-    "${GE_CODE_DIR}/metadef/graph/format_refiner.cc"
-    "${GE_CODE_DIR}/metadef/graph/inference_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/dumper/ge_graph_dumper.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/ge_ir_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/tensor_utils.cc"
-    "${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
-    "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
-    "${GE_CODE_DIR}/metadef/graph/runtime_inference_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/ref_relation.cc"
-    "${GE_CODE_DIR}/metadef/third_party/transformer/src/transfer_shape_according_to_format.cc"
-    "${GE_CODE_DIR}/metadef/third_party/transformer/src/axis_util.cc"
-    "${GE_CODE_DIR}/metadef/third_party/transformer/src/expand_dimension.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/transformer_utils.cc"
-)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph GRAPH_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/detail GRAPH_DETAIL_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/opsproto GRAPH_OPSPROTO_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/option GRAPH_OPTION_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/utils GRAPH_UTILS_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/utils/dumper GRAPH_DUMPER_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/ops GRAPH_OPS_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/third_party/transformer/src TRANSFORMER_SRC_FILES)
 
-#add_executable(ut_libgraph ${UT_FILES} ${SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
-add_executable(ut_libgraph ${UT_FILES} ${SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
+add_executable(ut_libgraph ${UT_FILES}
+    ${GRAPH_SRC_FILES}
+    ${GRAPH_UTILS_SRC_FILES}
+    ${GRAPH_OPSPROTO_SRC_FILES}
+    ${GRAPH_OPTION_SRC_FILES}
+    ${GRAPH_DETAIL_SRC_FILES}
+    ${GRAPH_DUMPER_SRC_FILES}
+    ${GRAPH_OPS_SRC_FILES}
+    ${TRANSFORMER_SRC_FILES}
+    ${PROTO_SRCS} ${PROTO_HDRS}
+)
 
 target_compile_options(ut_libgraph PRIVATE
     -g --coverage -fprofile-arcs -ftest-coverage

tests/ut/ge/CMakeLists.txt

@@ -20,6 +20,7 @@ set(CMAKE_CXX_STANDARD 11)
 set(PROTO_LIST
     "${GE_CODE_DIR}/metadef/proto/om.proto"
     "${GE_CODE_DIR}/metadef/proto/ge_ir.proto"
+    "${GE_CODE_DIR}/metadef/proto/task.proto"
     "${GE_CODE_DIR}/metadef/proto/ge_api.proto"
     "${GE_CODE_DIR}/metadef/proto/insert_op.proto"
     "${GE_CODE_DIR}/metadef/proto/dump_task.proto"
@@ -69,62 +70,19 @@ include_directories(${CMAKE_BINARY_DIR})
 include_directories(${CMAKE_BINARY_DIR}/proto/ge)
 include_directories(${CMAKE_BINARY_DIR}/proto/ge/proto)
 
-set(GRAPH_SRC_FILES
-    "${GE_CODE_DIR}/metadef/graph/option/ge_local_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/option/ge_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_attr_define.cc"
-    "${GE_CODE_DIR}/metadef/graph/anchor.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_attr_value.cc"
-    "${GE_CODE_DIR}/metadef/graph/attr_value.cc"
-    "${GE_CODE_DIR}/metadef/graph/buffer.cc"
-    "${GE_CODE_DIR}/metadef/graph/aligned_ptr.cc"
-    "${GE_CODE_DIR}/metadef/graph/compute_graph.cc"
-    "${GE_CODE_DIR}/metadef/graph/graph.cc"
-    "${GE_CODE_DIR}/metadef/graph/gnode.cc"
-    "${GE_CODE_DIR}/metadef/graph/ascend_string.cc"
-    "${GE_CODE_DIR}/metadef/graph/inference_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/shape_refiner.cc"
-    "${GE_CODE_DIR}/metadef/graph/model.cc"
-    "${GE_CODE_DIR}/metadef/graph/model_serialize.cc"
-    "${GE_CODE_DIR}/metadef/graph/node.cc"
-    "${GE_CODE_DIR}/metadef/graph/runtime_inference_context.cc"
-    "${GE_CODE_DIR}/metadef/graph/op_desc.cc"
-    "${GE_CODE_DIR}/metadef/third_party/transformer/src/transfer_shape_according_to_format.cc"
-    "${GE_CODE_DIR}/metadef/third_party/transformer/src/axis_util.cc"
-    "${GE_CODE_DIR}/metadef/third_party/transformer/src/expand_dimension.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
-    "${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"
-    "${GE_CODE_DIR}/metadef/graph/ge_tensor.cc"
-    "${GE_CODE_DIR}/metadef/graph/ref_relation.cc"
-    "${GE_CODE_DIR}/metadef/graph/tensor.cc"
-    "${GE_CODE_DIR}/metadef/graph/types.cc"
-    "${GE_CODE_DIR}/metadef/graph/detail/attributes_holder.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/anchor_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/graph_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/dumper/ge_graph_dumper.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/ge_ir_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/node_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/op_desc_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/tensor_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/type_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/transformer_utils.cc"
-    "${GE_CODE_DIR}/metadef/graph/debug/graph_debug.cc"
-    "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
-    "${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
-    "${GE_CODE_DIR}/metadef/register/register.cpp"
-    "${GE_CODE_DIR}/metadef/register/register_pass.cpp"
-    "${GE_CODE_DIR}/metadef/register/op_kernel_registry.cpp"
-    "${GE_CODE_DIR}/metadef/register/auto_mapping_util.cpp"
-    "${GE_CODE_DIR}/metadef/register/tensor_assign.cpp"
-    "${GE_CODE_DIR}/metadef/register/register_format_transfer.cc"
-    "${GE_CODE_DIR}/metadef/graph/format_refiner.cc"
-    "${GE_CODE_DIR}/metadef/register/ops_kernel_builder_registry.cc"
-    "${GE_CODE_DIR}/metadef/register/op_tiling.cpp"
-    "${GE_CODE_DIR}/metadef/graph/utils/tuning_utils.cc"
-    "${GE_CODE_DIR}/metadef/register/op_tiling_registry.cpp"
-    "${GE_CODE_DIR}/metadef/register/op_tiling_registry_impl.cpp"
-)
+
+#### GRAPH_SRC_FILES ####
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph GRAPH_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/debug GRAPH_DEBUG_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/detail GRAPH_DETAIL_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/opsproto GRAPH_OPSPROTO_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/option GRAPH_OPTION_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/utils GRAPH_UTILS_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/graph/utils/dumper GRAPH_DUMPER_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/ops GRAPH_OPS_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/register GRAPH_REGISTER_SRC_FILES)
+AUX_SOURCE_DIRECTORY(${GE_CODE_DIR}/metadef/third_party/transformer/src TRANSFORMER_SRC_FILES)
+
 
 set(PARSER_SRC_FILES
     "${GE_CODE_DIR}/parser/parser/common/op_map.cc"
@@ -259,7 +217,6 @@ set(GRAPH_PREPARE_COMMON_SRC_FILES
     "${GE_CODE_DIR}/ge/graph/preprocess/insert_op/util_insert_aipp_op.cc"
     "${GE_CODE_DIR}/ge/graph/preprocess/insert_op/ge_aipp_op.cc"
     "${GE_CODE_DIR}/ge/graph/preprocess/multi_batch_options.cc"
-    #"${GE_CODE_DIR}/ge/graph/preprocess/insert_op/base_insert_op.cc"
 )
 
 set(GRAPH_DAVINCI_MODEL_SRC_FILES
@@ -360,7 +317,6 @@ set(GRAPH_PASS_COMMON_SRC_FILES
     "${GE_CODE_DIR}/ge/graph/passes/compile_nodes_pass.cc"
     "${GE_CODE_DIR}/ge/graph/passes/flow_ctrl_pass.cc"
     "${GE_CODE_DIR}/ge/graph/passes/parallel_group_pass.cc"
-    #"${GE_CODE_DIR}/ge/graph/optimize/optimizer/allreduce_fusion_pass.cc"
     "${GE_CODE_DIR}/ge/graph/passes/folding_pass.cc"
     "${GE_CODE_DIR}/ge/graph/passes/variable_op_pass.cc"
     "${GE_CODE_DIR}/ge/graph/passes/transpose_transdata_pass.cc"
@@ -570,7 +526,6 @@ set(DISTINCT_GRAPH_LOAD_TEST_FILES
     #"graph/load/new_model_manager_data_inputer_unittest.cc"
     #"graph/load/new_model_manager_davinci_model_unittest.cc"
     "graph/load/model_manager_unittest.cc"
-    #"graph/load/new_model_manager_task_build_unittest.cc"
     "graph/load/new_model_manager_model_manager_aicpu_unittest.cc"
     "graph/load/end_graph_task_unittest.cc"
     "graph/load/new_model_manager_event_manager_unittest.cc"
@@ -795,7 +750,6 @@ set(OTHERS_TEST_FILES
 list(APPEND COMMON_SHARED_LIBRARIES
     c_sec
     slog_stub
-    cce_ge_stub
     runtime_stub
     profiler_stub
     mmpa_stub
@@ -808,7 +762,17 @@ list(APPEND COMMON_SHARED_LIBRARIES
 
 # build graph
 add_library(ge_ut_graph STATIC
-    ${GRAPH_SRC_FILES} ${PARSER_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}
+    ${GRAPH_SRC_FILES}
+    ${GRAPH_DEBUG_SRC_FILES}
+    ${GRAPH_DETAIL_SRC_FILES}
+    ${GRAPH_OPSPROTO_SRC_FILES}
+    ${GRAPH_OPTION_SRC_FILES}
+    ${GRAPH_UTILS_SRC_FILES}
+    ${GRAPH_DUMPER_SRC_FILES}
+    ${GRAPH_OPS_SRC_FILES}
+    ${GRAPH_REGISTER_SRC_FILES}
+    ${TRANSFORMER_SRC_FILES}
+    ${PARSER_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}
 )
 
 target_compile_definitions(ge_ut_graph PRIVATE