Browse Source

ge ut 2

tags/v1.1.0
t00456437 3 years ago
parent
commit
b2d1ab954b
28 changed files with 268 additions and 64 deletions
  1. +16
    -2
      tests/depends/cce/CMakeLists.txt
  2. +8
    -4
      tests/depends/hccl/CMakeLists.txt
  3. +9
    -9
      tests/depends/hccl/src/hccl_stub.cc
  4. +13
    -5
      tests/depends/mmpa/CMakeLists.txt
  5. +10
    -1
      tests/depends/omg/CMakeLists.txt
  6. +3
    -1
      tests/depends/omg/src/omg_stub.cc
  7. +7
    -3
      tests/depends/profiler/CMakeLists.txt
  8. +8
    -3
      tests/depends/runtime/CMakeLists.txt
  9. +4
    -1
      tests/depends/slog/CMakeLists.txt
  10. +113
    -2
      tests/ut/ge/CMakeLists.txt
  11. +1
    -1
      tests/ut/ge/common/datatype_transfer_unittest.cc
  12. +4
    -3
      tests/ut/ge/common/format_transfer_5d_nchw_unittest.cc
  13. +5
    -2
      tests/ut/ge/common/format_transfer_5d_nhwc_unittest.cc
  14. +5
    -2
      tests/ut/ge/common/format_transfer_c1hwncoc0_hwcn_unittest.cc
  15. +5
    -2
      tests/ut/ge/common/format_transfer_fractal_nz_unittest.cc
  16. +5
    -2
      tests/ut/ge/common/format_transfer_fractal_zz_unittest.cc
  17. +4
    -1
      tests/ut/ge/common/format_transfer_fracz_hwcn_unittest.cc
  18. +5
    -2
      tests/ut/ge/common/format_transfer_fracz_nchw_unittest.cc
  19. +5
    -2
      tests/ut/ge/common/format_transfer_fracz_nhwc_unittest.cc
  20. +5
    -2
      tests/ut/ge/common/format_transfer_hwcn_c1hwncoc0_unittest.cc
  21. +5
    -2
      tests/ut/ge/common/format_transfer_hwcn_fractalz_unittest.cc
  22. +4
    -1
      tests/ut/ge/common/format_transfer_nchw_5d_unittest.cc
  23. +5
    -2
      tests/ut/ge/common/format_transfer_nchw_fractalz_unittest.cc
  24. +5
    -2
      tests/ut/ge/common/format_transfer_nhwc_5d_unittest.cc
  25. +5
    -2
      tests/ut/ge/common/format_transfer_nhwc_fractalz_unittest.cc
  26. +6
    -2
      tests/ut/ge/common/format_transfer_unittest.cc
  27. +1
    -1
      tests/ut/ge/graph/build/logical_stream_allocator_unittest.cc
  28. +2
    -2
      tests/ut/ge/graph/build/mem_assigner_unittest.cc

+ 16
- 2
tests/depends/cce/CMakeLists.txt View File

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

+ 8
- 4
tests/depends/hccl/CMakeLists.txt View File

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

+ 9
- 9
tests/depends/hccl/src/hccl_stub.cc View File

@@ -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;
}
}

+ 13
- 5
tests/depends/mmpa/CMakeLists.txt View File

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

+ 10
- 1
tests/depends/omg/CMakeLists.txt View File

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

+ 3
- 1
tests/depends/omg/src/omg_stub.cc View File

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


+ 7
- 3
tests/depends/profiler/CMakeLists.txt View File

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

+ 8
- 3
tests/depends/runtime/CMakeLists.txt View File

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

+ 4
- 1
tests/depends/slog/CMakeLists.txt View File

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

+ 113
- 2
tests/ut/ge/CMakeLists.txt View File

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


+ 1
- 1
tests/ut/ge/common/datatype_transfer_unittest.cc View File

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



+ 4
- 3
tests/ut/ge/common/format_transfer_5d_nchw_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_5d_nhwc_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_c1hwncoc0_hwcn_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_fractal_nz_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_fractal_zz_unittest.cc View File

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

+ 4
- 1
tests/ut/ge/common/format_transfer_fracz_hwcn_unittest.cc View File

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


+ 5
- 2
tests/ut/ge/common/format_transfer_fracz_nchw_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_fracz_nhwc_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_hwcn_c1hwncoc0_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_hwcn_fractalz_unittest.cc View File

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

+ 4
- 1
tests/ut/ge/common/format_transfer_nchw_5d_unittest.cc View File

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


+ 5
- 2
tests/ut/ge/common/format_transfer_nchw_fractalz_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_nhwc_5d_unittest.cc View File

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

+ 5
- 2
tests/ut/ge/common/format_transfer_nhwc_fractalz_unittest.cc View File

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

+ 6
- 2
tests/ut/ge/common/format_transfer_unittest.cc View File

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

+ 1
- 1
tests/ut/ge/graph/build/logical_stream_allocator_unittest.cc View File

@@ -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);


+ 2
- 2
tests/ut/ge/graph/build/mem_assigner_unittest.cc View File

@@ -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(){};



Loading…
Cancel
Save