| @@ -23,8 +23,10 @@ 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) | |||
| @@ -66,7 +68,19 @@ set(SRCS | |||
| "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc" | |||
| ) | |||
| add_library(cce_ge_stub SHARED src/cce_stub.cc ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_link_libraries(cce_ge_stub ascend_protobuf) | |||
| 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_link_libraries(cce_stub ascend_protobuf) | |||
| target_link_libraries(cce_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| -Wl,--no-as-needed | |||
| ascend_protobuf | |||
| -Wl,--as-needed | |||
| c_sec | |||
| ) | |||
| @@ -13,14 +13,18 @@ | |||
| # limitations under the License. | |||
| # ============================================================================ | |||
| cmake_minimum_required(VERSION 2.8) | |||
| #cmake_minimum_required(VERSION 2.8) | |||
| project(hccl_stub) | |||
| file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | |||
| "src/hccl_stub.cc" | |||
| ) | |||
| include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_SOURCE_DIR}/inc) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_CODE_DIR}/inc) | |||
| add_library(hccl_stub SHARED ${SRC_FILES}) | |||
| add_library(hccl_stub SHARED ${SRC_FILES}) | |||
| target_link_libraries(hccl_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| ) | |||
| @@ -18,27 +18,27 @@ | |||
| #include "hccl/hcom.h" | |||
| hcclResult_t hcom_all_gather(const char *tag, void *input_count_ptr, void *output_ptr, u64 input_count, | |||
| hcclDataType_t data_type, const char *group, rtStream_t stream) { | |||
| HcclResult hcom_all_gather(const char *tag, void *input_count_ptr, void *output_ptr, u64 input_count, | |||
| HcclDataType data_type, const char *group, rtStream_t stream) { | |||
| return HCCL_SUCCESS; | |||
| } | |||
| hcclResult_t hcom_broadcast(const char *tag, void *ptr, u64 count, hcclDataType_t data_type, u32 root, | |||
| HcclResult hcom_broadcast(const char *tag, void *ptr, u64 count, HcclDataType data_type, u32 root, | |||
| const char *group, rtStream_t stream) { | |||
| return HCCL_SUCCESS; | |||
| } | |||
| hcclResult_t hcom_all_reduce(const char *tag, void *input_ptr, void *output_ptr, u64 count, hcclDataType_t data_type, | |||
| hcclRedOp_t op, const char *group, rtStream_t stream) { | |||
| HcclResult hcom_all_reduce(const char *tag, void *input_ptr, void *output_ptr, u64 count, HcclDataType data_type, | |||
| HcclReduceOp op, const char *group, rtStream_t stream) { | |||
| return HCCL_SUCCESS; | |||
| } | |||
| hcclResult_t hcom_get_split_strategy(const char *group, const struct model_feature *feature, u32 max_segment_num, | |||
| HcclResult hcom_get_split_strategy(const char *group, const struct model_feature *feature, u32 max_segment_num, | |||
| u32 *segment_num, u32 *segment_idx) { | |||
| return HCCL_SUCCESS; | |||
| } | |||
| hcclResult_t hcom_reduce_scatter(const char *tag, void *input_ptr, void *output_ptr, u64 count, | |||
| hcclDataType_t data_type, hcclRedOp_t op, const char *group, rtStream_t stream) { | |||
| HcclResult hcom_reduce_scatter(const char *tag, void *input_ptr, void *output_ptr, u64 count, | |||
| HcclDataType data_type, HcclReduceOp op, const char *group, rtStream_t stream) { | |||
| return HCCL_SUCCESS; | |||
| } | |||
| } | |||
| @@ -21,10 +21,18 @@ file(GLOB_RECURSE SRCS RELATIVE ${CMAKE_CURRENT_LIST_DIR} | |||
| "src/mmpa_stub.cc" | |||
| ) | |||
| include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_SOURCE_DIR}/inc) | |||
| include_directories(${GE_SOURCE_DIR}/inc/framework) | |||
| include_directories(${GE_SOURCE_DIR}/inc/external) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_CODE_DIR}/inc) | |||
| include_directories(${GE_CODE_DIR}/inc/external) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc) | |||
| include_directories(${GE_CODE_DIR}/inc/framework) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc/external) | |||
| add_library(mmpa_stub SHARED ${SRCS}) | |||
| target_link_libraries(mmpa_stub ascend_protobuf) | |||
| target_link_libraries(mmpa_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| -Wl,--no-as-needed | |||
| ascend_protobuf | |||
| -Wl,--as-needed | |||
| c_sec | |||
| ) | |||
| @@ -22,9 +22,11 @@ set(CMAKE_CXX_STANDARD 11) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce) | |||
| include_directories(${GE_CODE_DIR}/inc) | |||
| include_directories(${GE_CODE_DIR}/metadef/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}/ge) | |||
| include_directories(${CMAKE_BINARY_DIR}) | |||
| @@ -42,4 +44,11 @@ set(SRCS | |||
| ) | |||
| add_library(omg_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_link_libraries(omg_stub ascend_protobuf) | |||
| target_link_libraries(omg_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| -Wl,--no-as-needed | |||
| ascend_protobuf | |||
| -Wl,--as-needed | |||
| c_sec | |||
| json | |||
| ) | |||
| @@ -643,7 +643,7 @@ Status GetInputOutputDescInfo(uint32_t model_id, vector<InputOutputDescInfo> &in | |||
| } | |||
| Status DataInput(const InputData *input_data, OutputData *output_data) { return SUCCESS; } | |||
| /* | |||
| class ModelManager { | |||
| public: | |||
| static std::shared_ptr<ModelManager> GetInstance(); | |||
| @@ -741,6 +741,8 @@ Status ModelManager::ExecuteModel(uint32_t model_id, rtStream_t stream, bool asy | |||
| return SUCCESS; | |||
| } | |||
| */ | |||
| } // namespace ge | |||
| namespace ge { | |||
| @@ -13,12 +13,16 @@ | |||
| # limitations under the License. | |||
| # ============================================================================ | |||
| cmake_minimum_required(VERSION 2.8) | |||
| #cmake_minimum_required(VERSION 2.8) | |||
| project(profiler_stub) | |||
| file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | |||
| "src/profiler_stub.cc" | |||
| ) | |||
| include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | |||
| add_library(profiler_stub SHARED ${SRC_FILES}) | |||
| add_library(profiler_stub SHARED ${SRC_FILES}) | |||
| target_link_libraries(profiler_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| ) | |||
| @@ -13,7 +13,7 @@ | |||
| # limitations under the License. | |||
| # ============================================================================ | |||
| cmake_minimum_required(VERSION 2.8) | |||
| #cmake_minimum_required(VERSION 2.8) | |||
| project(STUB_MMPA) | |||
| @@ -21,7 +21,12 @@ file(GLOB_RECURSE SRCS RELATIVE ${CMAKE_CURRENT_LIST_DIR} | |||
| "src/runtime_stub.cc" | |||
| ) | |||
| include_directories(${GE_SOURCE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_SOURCE_DIR}/inc/framework) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | |||
| include_directories(${GE_CODE_DIR}/inc/framework) | |||
| add_library(runtime_stub SHARED ${SRCS}) | |||
| target_link_libraries(runtime_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| c_sec | |||
| ) | |||
| @@ -13,7 +13,7 @@ | |||
| # limitations under the License. | |||
| # ============================================================================ | |||
| cmake_minimum_required(VERSION 2.8) | |||
| #cmake_minimum_required(VERSION 2.8) | |||
| project(slog_stub) | |||
| file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | |||
| @@ -21,3 +21,6 @@ file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | |||
| ) | |||
| include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | |||
| add_library(slog_stub SHARED ${SRC_FILES}) | |||
| target_link_libraries(slog_stub PRIVATE | |||
| $<BUILD_INTERFACE:intf_pub> | |||
| ) | |||
| @@ -23,7 +23,14 @@ set(PROTO_LIST | |||
| "${GE_CODE_DIR}/metadef/proto/ge_api.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/fwk_adapter.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/op_mapping_info.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/optimizer_priority.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/ge_api.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/tensorflow/attr_value.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/tensorflow/tensor.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/tensorflow/resource_handle.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/tensorflow/tensor_shape.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/tensorflow/types.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/tensorflow/node_def.proto" | |||
| "${GE_CODE_DIR}/metadef/proto/proto_inner/ge_onnx.proto" | |||
| ) | |||
| @@ -33,17 +40,22 @@ protobuf_generate(ge PROTO_SRCS PROTO_HDRS ${PROTO_LIST}) | |||
| include_directories(${CMAKE_CURRENT_LIST_DIR}) | |||
| include_directories(${GE_CODE_DIR}) | |||
| include_directories(${GE_CODE_DIR}/inc) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc) | |||
| include_directories(${GE_CODE_DIR}/ge) | |||
| include_directories(${GE_CODE_DIR}/metadef) | |||
| include_directories(${GE_CODE_DIR}/metadef/graph) | |||
| include_directories(${GE_CODE_DIR}/inc/external) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc/external) | |||
| include_directories(${GE_CODE_DIR}/parser/parser) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc/external/graph) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc/graph) | |||
| include_directories(${GE_CODE_DIR}/inc/framework) | |||
| include_directories(${GE_CODE_DIR}/metadef/inc/common) | |||
| include_directories(${GE_CODE_DIR}/metadef/third_party) | |||
| 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(${GE_CODE_DIR}/third_party/fwkacllib/inc/toolchain) | |||
| include_directories(${GE_CODE_DIR}/tests/ut/ge) | |||
| include_directories(${CMAKE_BINARY_DIR}) | |||
| include_directories(${CMAKE_BINARY_DIR}/proto/ge) | |||
| @@ -501,33 +513,127 @@ list(APPEND COMMON_SHARED_LIBRARIES | |||
| # build common | |||
| add_library(ge_ut_common STATIC ${COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_ut_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_ut_common PRIVATE | |||
| c_sec | |||
| ascend_protobuf | |||
| json | |||
| ) | |||
| # build common format | |||
| add_library(ge_ut_common_format STATIC ${COMMON_SRC_FILES} ${COMMON_FORMAT_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_ut_common_format PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| # build graph prepare common | |||
| add_library(ge_prepare_common STATIC ${GRAPH_PREPARE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_prepare_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_prepare_common PRIVATE | |||
| c_sec | |||
| ascend_protobuf | |||
| json | |||
| ) | |||
| # build graph optimize common | |||
| add_library(ge_optimize_common STATIC ${GRAPH_OPTIMIZE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_optimize_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_optimize_common PRIVATE | |||
| ascend_protobuf | |||
| c_sec | |||
| json | |||
| ) | |||
| # build graph partition common | |||
| add_library(ge_partition_common STATIC ${GRAPH_PARTITION_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_partition_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_partition_common PRIVATE | |||
| ascend_protobuf | |||
| c_sec | |||
| json | |||
| ) | |||
| # build build graph load common | |||
| add_library(ge_load_common STATIC ${GRAPH_LOAD_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_load_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_load_common PRIVATE | |||
| c_sec | |||
| ascend_protobuf | |||
| json | |||
| ) | |||
| # build graph execute common | |||
| add_library(ge_execute_common STATIC ${GRAPH_EXECUTE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_execute_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_execute_common PRIVATE | |||
| c_sec | |||
| json | |||
| ascend_protobuf | |||
| ) | |||
| # build graph build common | |||
| add_library(ge_build_common STATIC ${GRAPH_BUILD_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_build_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_build_common PRIVATE | |||
| c_sec | |||
| ascend_protobuf | |||
| json | |||
| ) | |||
| # build graph pass common | |||
| add_library(ge_pass_common STATIC ${GRAPH_PASS_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_pass_common PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_pass_common PRIVATE | |||
| ascend_protobuf | |||
| c_sec | |||
| json | |||
| ) | |||
| # build single_op common | |||
| add_library(ge_single_op STATIC ${SINGLE_OP_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | |||
| target_compile_definitions(ge_single_op PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ge_single_op PRIVATE | |||
| ascend_protobuf | |||
| json | |||
| c_sec | |||
| ) | |||
| # ut binary | |||
| # libge_mutiparts_utest | |||
| @@ -536,12 +642,17 @@ add_executable(ut_libge_multiparts_utest | |||
| ${COMMON_FORMAT_SRC_FILES} | |||
| ${MULTI_PARTS_TEST_FILES} | |||
| ) | |||
| target_compile_definitions(ut_libge_multiparts_utest PRIVATE | |||
| google=ascend_private | |||
| ) | |||
| target_link_libraries(ut_libge_multiparts_utest | |||
| ge_build_common ge_load_common ge_build_common ge_execute_common ge_optimize_common ge_partition_common ge_pass_common | |||
| ge_prepare_common ge_single_op ge_ut_common | |||
| gtest gtest_main ascend_protobuf -lrt -ldl | |||
| gtest gtest_main ascend_protobuf -lrt -ldl | |||
| ) | |||
| target_link_libraries(ut_libge_multiparts_utest ${COMMON_SHARED_LIBRARIES} ascend_protobuf) | |||
| target_link_libraries(ut_libge_multiparts_utest ${COMMON_SHARED_LIBRARIES} ascend_protobuf json) | |||
| # libge_others_utest | |||
| add_executable(ut_libge_others_utest | |||
| @@ -18,7 +18,7 @@ | |||
| #include "common/formats/format_transfers/datatype_transfer.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/formats/formats.h" | |||
| #include "common/fp16_t.h" | |||
| @@ -17,9 +17,10 @@ | |||
| #include <gtest/gtest.h> | |||
| #include "common/formats/format_transfers/format_transfer_nc1hwc0_nchw.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| @@ -644,4 +645,4 @@ TEST_F(UTEST_FormatTransferNc1hwc0ToNchw, invalid_src_data_type) { | |||
| EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_nc1hwc0_nhwc.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransfer5dNhwc : public testing::Test { | |||
| @@ -759,4 +762,4 @@ TEST_F(UtestFormatTransfer5dNhwc, invalid_src_dst_shape_relation) { | |||
| EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_c1hwncoc0_hwcn.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferC1hwncoc0Hwcn : public testing::Test { | |||
| @@ -13710,4 +13713,4 @@ TEST_F(UtestFormatTransferC1hwncoc0Hwcn, sixd_to_hwcn_fp32_success_gt_cube) { | |||
| } | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -19,11 +19,14 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fractal_nz.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/formats/formats.h" | |||
| #include "common/fp16_t.h" | |||
| #include "time.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferNdFractNz : public testing::Test { | |||
| @@ -9164,4 +9167,4 @@ TEST_F(UtestFormatTransferNdFractNz, invalid_src_dst_shape_relation) { | |||
| EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -19,11 +19,14 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fractal_zz.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/formats/formats.h" | |||
| #include "common/fp16_t.h" | |||
| #include "time.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferNdFractZz : public testing::Test { | |||
| @@ -7988,4 +7991,4 @@ TEST_F(UtestFormatTransferNdFractZz, invalid_src_dst_shape_relation) { | |||
| EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fracz_hwcn.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferFracZHwcn : public testing::Test { | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fracz_nchw.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferFraczNchw : public testing::Test { | |||
| @@ -10486,4 +10489,4 @@ TEST_F(UtestFormatTransferFraczNchw, fp32_1) { | |||
| } | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fracz_nhwc.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferFraczNhwc : public testing::Test { | |||
| @@ -5422,4 +5425,4 @@ TEST_F(UtestFormatTransferFraczNhwc, fracz_to_nhwc_fp32_success_gt_cube) { | |||
| } | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_hwcn_c1hwncoc0.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferHwcnC1hwncoc0 : public testing::Test { | |||
| @@ -13745,4 +13748,4 @@ TEST_F(UtestFormatTransferHwcnC1hwncoc0, hwcn_to_6d_fp32_success_gt_cube) { | |||
| } | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,7 +18,10 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fractal_z.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| @@ -34460,4 +34463,4 @@ TEST_F(UtestFormatTransferHwcnFz, build_transfer_not_support) { | |||
| EXPECT_EQ(transfer, nullptr); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,7 +18,10 @@ | |||
| #include "common/formats/format_transfers/format_transfer_nchw_nc1hwc0.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| @@ -18,7 +18,10 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fractal_z.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| @@ -16873,4 +16876,4 @@ TEST_F(UtestFormatTransferNchwFz, build_transfer_uint8) { | |||
| EXPECT_NE(transfer, nullptr); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,12 @@ | |||
| #include "common/formats/format_transfers/format_transfer_nhwc_nc1hwc0.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/fp16_t.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| class UtestFormatTransferNhwc5d : public testing::Test { | |||
| @@ -747,4 +750,4 @@ TEST_F(UtestFormatTransferNhwc5d, unsupport_dst_format) { | |||
| EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,7 +18,10 @@ | |||
| #include "common/formats/format_transfers/format_transfer_fractal_z.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| @@ -5351,4 +5354,4 @@ TEST_F(UtestFormatTransferNhwcFz, build_transfer_uint8) { | |||
| EXPECT_NE(transfer, nullptr); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -18,9 +18,13 @@ | |||
| #include "common/formats/format_transfers/format_transfer_nchw_nc1hwc0.h" | |||
| #include "common/formats/format_transfers/format_transfer.h" | |||
| //#include "common/formats/format_transfers/format_transfer.h" | |||
| #include "common/formats/utils/formats_trans_utils.h" | |||
| #include "register/register_format_transfer.h" | |||
| #include "framework/common/ge_inner_error_codes.h" | |||
| namespace ge { | |||
| namespace formats { | |||
| @@ -78,4 +82,4 @@ TEST_F(UtestFormatTransfer, get_size_by_data_type) { | |||
| EXPECT_EQ(DT_UNDEFINED, 26); | |||
| } | |||
| } // namespace formats | |||
| } // namespace ge | |||
| } // namespace ge | |||
| @@ -858,7 +858,7 @@ TEST_F(UtestLogicalStreamAllocator, test_all_reduce_parallel_pass) { | |||
| std::map<std::string, int> max_parallel_num; | |||
| LogicalStreamPass::Context context; | |||
| context.next_stream = 5; | |||
| context.hcom_parallel = true; | |||
| context.enable_hcom_parallel = true; | |||
| vector<LogicalStreamPass::SubgraphPtr> subgraphs; | |||
| LogicalStreamPassPtr allreduce_pass = std::make_shared<AllReduceParallelPass>(); | |||
| ret = allreduce_pass->Run(graph, subgraphs, context); | |||
| @@ -152,7 +152,7 @@ TEST_F(UtestMemoryAssignerTest, MemoryBlock_Resize_RealSizeList_is_empty) { | |||
| ge::OpDescPtr op_def_a = createOpWithWsSize("A", 6000); | |||
| ge::NodePtr node_a = graph->AddNode(op_def_a); | |||
| MemoryBlock* memory_block = new MemoryBlock(0); | |||
| memory_block->Init(1, kOutput, node_a, 0); | |||
| memory_block->Init(1, kOutput, node_a, 0, 1); | |||
| memory_block->real_size_list_.clear(); | |||
| memory_block->Resize(); | |||
| @@ -165,7 +165,7 @@ namespace ge { | |||
| class MockBlockMemAssigner : public BlockMemAssigner { | |||
| public: | |||
| explicit MockBlockMemAssigner(ge::ComputeGraphPtr compute_graph) : BlockMemAssigner(compute_graph){}; | |||
| explicit MockBlockMemAssigner(ge::ComputeGraphPtr compute_graph, const std::map<std::string, std::string> &anchor_to_symbol, const std::map<std::string, std::list<NodeIndexIO>> &symbol_to_anchors) : BlockMemAssigner(compute_graph, anchor_to_symbol, symbol_to_anchors) {}; | |||
| virtual ~MockBlockMemAssigner(){}; | |||