@@ -23,8 +23,10 @@ include_directories(${GE_CODE_DIR}/inc) | |||||
include_directories(${GE_CODE_DIR}/inc/framework) | include_directories(${GE_CODE_DIR}/inc/framework) | ||||
include_directories(${GE_CODE_DIR}/metadef/inc/graph) | include_directories(${GE_CODE_DIR}/metadef/inc/graph) | ||||
include_directories(${GE_CODE_DIR}/inc/external) | 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/inc/external/graph) | ||||
include_directories(${GE_CODE_DIR}/metadef) | 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}/metadef/graph) | ||||
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | 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/cce) | ||||
@@ -66,7 +68,19 @@ set(SRCS | |||||
"${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc" | "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc" | ||||
) | ) | ||||
add_library(cce_ge_stub SHARED src/cce_stub.cc ${PROTO_SRCS} ${PROTO_HDRS}) | 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}) | 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. | # limitations under the License. | ||||
# ============================================================================ | # ============================================================================ | ||||
cmake_minimum_required(VERSION 2.8) | |||||
#cmake_minimum_required(VERSION 2.8) | |||||
project(hccl_stub) | project(hccl_stub) | ||||
file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | ||||
"src/hccl_stub.cc" | "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" | #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; | 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) { | const char *group, rtStream_t stream) { | ||||
return HCCL_SUCCESS; | 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; | 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) { | u32 *segment_num, u32 *segment_idx) { | ||||
return HCCL_SUCCESS; | 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; | return HCCL_SUCCESS; | ||||
} | |||||
} |
@@ -21,10 +21,18 @@ file(GLOB_RECURSE SRCS RELATIVE ${CMAKE_CURRENT_LIST_DIR} | |||||
"src/mmpa_stub.cc" | "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}) | 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) | ||||
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce) | include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce) | ||||
include_directories(${GE_CODE_DIR}/inc) | 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}/inc/framework) | ||||
include_directories(${GE_CODE_DIR}/metadef/inc/graph) | include_directories(${GE_CODE_DIR}/metadef/inc/graph) | ||||
include_directories(${GE_CODE_DIR}/inc/external) | 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/inc/external/graph) | ||||
include_directories(${GE_CODE_DIR}/ge) | include_directories(${GE_CODE_DIR}/ge) | ||||
include_directories(${CMAKE_BINARY_DIR}) | include_directories(${CMAKE_BINARY_DIR}) | ||||
@@ -42,4 +44,11 @@ set(SRCS | |||||
) | ) | ||||
add_library(omg_stub SHARED ${SRCS} ${PROTO_SRCS} ${PROTO_HDRS}) | 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; } | Status DataInput(const InputData *input_data, OutputData *output_data) { return SUCCESS; } | ||||
/* | |||||
class ModelManager { | class ModelManager { | ||||
public: | public: | ||||
static std::shared_ptr<ModelManager> GetInstance(); | static std::shared_ptr<ModelManager> GetInstance(); | ||||
@@ -741,6 +741,8 @@ Status ModelManager::ExecuteModel(uint32_t model_id, rtStream_t stream, bool asy | |||||
return SUCCESS; | return SUCCESS; | ||||
} | } | ||||
*/ | |||||
} // namespace ge | } // namespace ge | ||||
namespace ge { | namespace ge { | ||||
@@ -13,12 +13,16 @@ | |||||
# limitations under the License. | # limitations under the License. | ||||
# ============================================================================ | # ============================================================================ | ||||
cmake_minimum_required(VERSION 2.8) | |||||
#cmake_minimum_required(VERSION 2.8) | |||||
project(profiler_stub) | project(profiler_stub) | ||||
file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | ||||
"src/profiler_stub.cc" | "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. | # limitations under the License. | ||||
# ============================================================================ | # ============================================================================ | ||||
cmake_minimum_required(VERSION 2.8) | |||||
#cmake_minimum_required(VERSION 2.8) | |||||
project(STUB_MMPA) | project(STUB_MMPA) | ||||
@@ -21,7 +21,12 @@ file(GLOB_RECURSE SRCS RELATIVE ${CMAKE_CURRENT_LIST_DIR} | |||||
"src/runtime_stub.cc" | "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}) | 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. | # limitations under the License. | ||||
# ============================================================================ | # ============================================================================ | ||||
cmake_minimum_required(VERSION 2.8) | |||||
#cmake_minimum_required(VERSION 2.8) | |||||
project(slog_stub) | project(slog_stub) | ||||
file(GLOB_RECURSE SRC_FILES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} | 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) | include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc) | ||||
add_library(slog_stub SHARED ${SRC_FILES}) | 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/ge_api.proto" | ||||
"${GE_CODE_DIR}/metadef/proto/fwk_adapter.proto" | "${GE_CODE_DIR}/metadef/proto/fwk_adapter.proto" | ||||
"${GE_CODE_DIR}/metadef/proto/op_mapping_info.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/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" | "${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(${CMAKE_CURRENT_LIST_DIR}) | ||||
include_directories(${GE_CODE_DIR}) | include_directories(${GE_CODE_DIR}) | ||||
include_directories(${GE_CODE_DIR}/inc) | include_directories(${GE_CODE_DIR}/inc) | ||||
include_directories(${GE_CODE_DIR}/metadef/inc) | |||||
include_directories(${GE_CODE_DIR}/ge) | include_directories(${GE_CODE_DIR}/ge) | ||||
include_directories(${GE_CODE_DIR}/metadef) | include_directories(${GE_CODE_DIR}/metadef) | ||||
include_directories(${GE_CODE_DIR}/metadef/graph) | include_directories(${GE_CODE_DIR}/metadef/graph) | ||||
include_directories(${GE_CODE_DIR}/inc/external) | 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/external/graph) | ||||
include_directories(${GE_CODE_DIR}/metadef/inc/graph) | include_directories(${GE_CODE_DIR}/metadef/inc/graph) | ||||
include_directories(${GE_CODE_DIR}/inc/framework) | include_directories(${GE_CODE_DIR}/inc/framework) | ||||
include_directories(${GE_CODE_DIR}/metadef/inc/common) | 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) | ||||
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/cce) | 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/ops) | ||||
include_directories(${GE_CODE_DIR}/third_party/fwkacllib/inc/toolchain) | |||||
include_directories(${GE_CODE_DIR}/tests/ut/ge) | include_directories(${GE_CODE_DIR}/tests/ut/ge) | ||||
include_directories(${CMAKE_BINARY_DIR}) | include_directories(${CMAKE_BINARY_DIR}) | ||||
include_directories(${CMAKE_BINARY_DIR}/proto/ge) | include_directories(${CMAKE_BINARY_DIR}/proto/ge) | ||||
@@ -501,33 +513,127 @@ list(APPEND COMMON_SHARED_LIBRARIES | |||||
# build common | # build common | ||||
add_library(ge_ut_common STATIC ${COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build common format | ||||
add_library(ge_ut_common_format STATIC ${COMMON_SRC_FILES} ${COMMON_FORMAT_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build graph prepare common | ||||
add_library(ge_prepare_common STATIC ${GRAPH_PREPARE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build graph optimize common | ||||
add_library(ge_optimize_common STATIC ${GRAPH_OPTIMIZE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build graph partition common | ||||
add_library(ge_partition_common STATIC ${GRAPH_PARTITION_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build build graph load common | ||||
add_library(ge_load_common STATIC ${GRAPH_LOAD_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build graph execute common | ||||
add_library(ge_execute_common STATIC ${GRAPH_EXECUTE_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build graph build common | ||||
add_library(ge_build_common STATIC ${GRAPH_BUILD_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build graph pass common | ||||
add_library(ge_pass_common STATIC ${GRAPH_PASS_COMMON_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # build single_op common | ||||
add_library(ge_single_op STATIC ${SINGLE_OP_SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS}) | 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 | # ut binary | ||||
# libge_mutiparts_utest | # libge_mutiparts_utest | ||||
@@ -536,12 +642,17 @@ add_executable(ut_libge_multiparts_utest | |||||
${COMMON_FORMAT_SRC_FILES} | ${COMMON_FORMAT_SRC_FILES} | ||||
${MULTI_PARTS_TEST_FILES} | ${MULTI_PARTS_TEST_FILES} | ||||
) | ) | ||||
target_compile_definitions(ut_libge_multiparts_utest PRIVATE | |||||
google=ascend_private | |||||
) | |||||
target_link_libraries(ut_libge_multiparts_utest | 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_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 | 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 | # libge_others_utest | ||||
add_executable(ut_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/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/formats/formats.h" | ||||
#include "common/fp16_t.h" | #include "common/fp16_t.h" | ||||
@@ -17,9 +17,10 @@ | |||||
#include <gtest/gtest.h> | #include <gtest/gtest.h> | ||||
#include "common/formats/format_transfers/format_transfer_nc1hwc0_nchw.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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
@@ -644,4 +645,4 @@ TEST_F(UTEST_FormatTransferNc1hwc0ToNchw, invalid_src_data_type) { | |||||
EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransfer5dNhwc : public testing::Test { | 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); | EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferC1hwncoc0Hwcn : public testing::Test { | class UtestFormatTransferC1hwncoc0Hwcn : public testing::Test { | ||||
@@ -13710,4 +13713,4 @@ TEST_F(UtestFormatTransferC1hwncoc0Hwcn, sixd_to_hwcn_fp32_success_gt_cube) { | |||||
} | } | ||||
} | } | ||||
} // namespace formats | } // 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_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/formats/formats.h" | ||||
#include "common/fp16_t.h" | #include "common/fp16_t.h" | ||||
#include "time.h" | #include "time.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferNdFractNz : public testing::Test { | 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); | EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | ||||
} | } | ||||
} // namespace formats | } // 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_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/formats/formats.h" | ||||
#include "common/fp16_t.h" | #include "common/fp16_t.h" | ||||
#include "time.h" | #include "time.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferNdFractZz : public testing::Test { | 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); | EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferFracZHwcn : public testing::Test { | 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferFraczNchw : public testing::Test { | class UtestFormatTransferFraczNchw : public testing::Test { | ||||
@@ -10486,4 +10489,4 @@ TEST_F(UtestFormatTransferFraczNchw, fp32_1) { | |||||
} | } | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferFraczNhwc : public testing::Test { | class UtestFormatTransferFraczNhwc : public testing::Test { | ||||
@@ -5422,4 +5425,4 @@ TEST_F(UtestFormatTransferFraczNhwc, fracz_to_nhwc_fp32_success_gt_cube) { | |||||
} | } | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferHwcnC1hwncoc0 : public testing::Test { | class UtestFormatTransferHwcnC1hwncoc0 : public testing::Test { | ||||
@@ -13745,4 +13748,4 @@ TEST_F(UtestFormatTransferHwcnC1hwncoc0, hwcn_to_6d_fp32_success_gt_cube) { | |||||
} | } | ||||
} | } | ||||
} // namespace formats | } // 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_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 ge { | ||||
namespace formats { | namespace formats { | ||||
@@ -34460,4 +34463,4 @@ TEST_F(UtestFormatTransferHwcnFz, build_transfer_not_support) { | |||||
EXPECT_EQ(transfer, nullptr); | EXPECT_EQ(transfer, nullptr); | ||||
} | } | ||||
} // namespace formats | } // 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_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 ge { | ||||
namespace formats { | namespace formats { | ||||
@@ -18,7 +18,10 @@ | |||||
#include "common/formats/format_transfers/format_transfer_fractal_z.h" | #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 ge { | ||||
namespace formats { | namespace formats { | ||||
@@ -16873,4 +16876,4 @@ TEST_F(UtestFormatTransferNchwFz, build_transfer_uint8) { | |||||
EXPECT_NE(transfer, nullptr); | EXPECT_NE(transfer, nullptr); | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/fp16_t.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
class UtestFormatTransferNhwc5d : public testing::Test { | class UtestFormatTransferNhwc5d : public testing::Test { | ||||
@@ -747,4 +750,4 @@ TEST_F(UtestFormatTransferNhwc5d, unsupport_dst_format) { | |||||
EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | EXPECT_EQ(transfer.TransFormat(args, result), PARAM_INVALID); | ||||
} | } | ||||
} // namespace formats | } // 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_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 ge { | ||||
namespace formats { | namespace formats { | ||||
@@ -5351,4 +5354,4 @@ TEST_F(UtestFormatTransferNhwcFz, build_transfer_uint8) { | |||||
EXPECT_NE(transfer, nullptr); | EXPECT_NE(transfer, nullptr); | ||||
} | } | ||||
} // namespace formats | } // 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_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 "common/formats/utils/formats_trans_utils.h" | ||||
#include "register/register_format_transfer.h" | |||||
#include "framework/common/ge_inner_error_codes.h" | |||||
namespace ge { | namespace ge { | ||||
namespace formats { | namespace formats { | ||||
@@ -78,4 +82,4 @@ TEST_F(UtestFormatTransfer, get_size_by_data_type) { | |||||
EXPECT_EQ(DT_UNDEFINED, 26); | EXPECT_EQ(DT_UNDEFINED, 26); | ||||
} | } | ||||
} // namespace formats | } // 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; | std::map<std::string, int> max_parallel_num; | ||||
LogicalStreamPass::Context context; | LogicalStreamPass::Context context; | ||||
context.next_stream = 5; | context.next_stream = 5; | ||||
context.hcom_parallel = true; | |||||
context.enable_hcom_parallel = true; | |||||
vector<LogicalStreamPass::SubgraphPtr> subgraphs; | vector<LogicalStreamPass::SubgraphPtr> subgraphs; | ||||
LogicalStreamPassPtr allreduce_pass = std::make_shared<AllReduceParallelPass>(); | LogicalStreamPassPtr allreduce_pass = std::make_shared<AllReduceParallelPass>(); | ||||
ret = allreduce_pass->Run(graph, subgraphs, context); | 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::OpDescPtr op_def_a = createOpWithWsSize("A", 6000); | ||||
ge::NodePtr node_a = graph->AddNode(op_def_a); | ge::NodePtr node_a = graph->AddNode(op_def_a); | ||||
MemoryBlock* memory_block = new MemoryBlock(0); | 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->real_size_list_.clear(); | ||||
memory_block->Resize(); | memory_block->Resize(); | ||||
@@ -165,7 +165,7 @@ namespace ge { | |||||
class MockBlockMemAssigner : public BlockMemAssigner { | class MockBlockMemAssigner : public BlockMemAssigner { | ||||
public: | 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(){}; | virtual ~MockBlockMemAssigner(){}; | ||||