p85620714
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

changing ops paths to cope with latest ascend packages 

support uint1

add uint1

update graphengine
This commit is contained in:
yanghaoran 2022-10-09 17:33:44 +08:00 committed by YangLuo
parent d542022a88
commit 9d391fb1a2
41 changed files with 126 additions and 95 deletions

6
cmake/ascend_variables.cmake
View File

@ -9,14 +9,14 @@ set(ASCEND_DRIVER_HAL_PATH ${ASCEND_PATH}/driver/lib64/driver)
# CANN packages
set(ASCEND_CANN_RUNTIME_PATH ${ASCEND_PATH}/latest/lib64)
set(ASCEND_CANN_OPP_PATH ${ASCEND_PATH}/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(ASCEND_CANN_OPP_PATH ${ASCEND_PATH}/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
set(ASCEND_CANN_PLUGIN_PATH ${ASCEND_CANN_RUNTIME_PATH}/plugin/opskernel)
# Ascend-toolkit packages
set(ASCEND_TOOLKIT_RUNTIME_PATH ${ASCEND_PATH}/ascend-toolkit/latest/lib64)
set(ASCEND_TOOLKIT_OPP_PATH ${ASCEND_PATH}/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(ASCEND_TOOLKIT_OPP_PATH ${ASCEND_PATH}/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
set(ASCEND_TOOLKIT_PLUGIN_PATH ${ASCEND_TOOLKIT_RUNTIME_PATH}/plugin/opskernel)
# nnae packages (for rpath only)
set(ASCEND_NNAE_RUNTIME_PATH ${ASCEND_PATH}/nnae/latest/lib64)
set(ASCEND_NNAE_OPP_PATH ${ASCEND_PATH}/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(ASCEND_NNAE_OPP_PATH ${ASCEND_PATH}/nnae/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)

2
graphengine

@ -1 +1 @@
Subproject commit 236001806129e36c0f48b240c4f61b2e1d92c470
Subproject commit 8ac63cd76d3988fa3ce84f227593e950c229041a

7
mindspore/ccsrc/cxx_api/CMakeLists.txt
View File

@ -205,12 +205,11 @@ if(CMAKE_SYSTEM_NAME MATCHES "Linux")
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/lib64)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/lib64)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/latest/lib64)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH
${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH
${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
endif()
set_target_properties(mindspore_shared_lib PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})
endif()

1
mindspore/ccsrc/kernel/common_utils.cc
View File

@ -94,6 +94,7 @@ const std::unordered_map<FusionType, std::string> fusion_type_name_maps = {
{FusionType::BATCH_MATMUL, "BatchMatmul"},
{FusionType::CONFUSION_TRANSPOSE, "confusiontranspose"},
{FusionType::DROPOUT_DOMASKV3D, "DropOutDoMaskV3D"},
{FusionType::GATHER_ASCEND, "Gather"},
{FusionType::UNKNOWN_FUSION_TYPE, ""}};
abstract::BaseShapePtr GetValidShapeFromAbstract(const abstract::AbstractBasePtr &abs) {

1
mindspore/ccsrc/kernel/kernel.h
View File

@ -100,6 +100,7 @@ enum FusionType {
BATCH_MATMUL,
CONFUSION_TRANSPOSE,
DROPOUT_DOMASKV3D,
GATHER_ASCEND,
UNKNOWN_FUSION_TYPE = -1,
};

8
mindspore/ccsrc/plugin/device/ascend/CMakeLists.txt
View File

@ -4,11 +4,11 @@ include_directories(${CMAKE_BINARY_DIR})
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/lib64)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/lib64)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/latest/lib64)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH
${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling)
${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling)
########### mindspore_ascend.so #####
set(ASCEND_SUB_COMP

4
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_kernel_build.cc
View File

@ -233,7 +233,7 @@ void SetNodeInputs(const std::shared_ptr<AnfNode> &anf_node, mindspore::NodeDef
auto value = GetValue<std::string>(value_ptr);
input_shape.push_back(1);
input_shape.push_back(static_cast<int64_t>(value.size()));
input_data_type = AicpuOpUtil::MsTypeToProtoType(kTypeUnknown);
input_data_type = AicpuOpUtil::MsTypeToProtoType(kObjectTypeString);
} else {
input_shape = AnfAlgo::GetInputDeviceShape(anf_node, input_index);
input_data_type = AicpuOpUtil::MsTypeToProtoType(input_type);
@ -360,7 +360,7 @@ uint64_t SetExtInfoInputShapeType(char *ext_info_buf, uint64_t ext_info_offset,
auto value = GetValue<std::string>(value_ptr);
input_shape.push_back(1);
input_shape.push_back(static_cast<int64_t>(value.size()));
input_data_type = AicpuOpUtil::MsTypeToProtoType(kTypeUnknown);
input_data_type = AicpuOpUtil::MsTypeToProtoType(kObjectTypeString);
} else {
input_shape = AnfAlgo::GetInputDeviceShape(anf_node, input_index);
input_data_type = AicpuOpUtil::MsTypeToProtoType(input_type);

2
mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_util.cc
View File

@ -38,7 +38,7 @@ static const std::map<int32_t, int32_t> kMsProtoDataTypeMap = {
{mindspore::TypeId::kNumberTypeFloat64, mindspore::DataType::MS_FLOAT64},
{mindspore::TypeId::kNumberTypeComplex64, mindspore::DataType::MS_COMPLEX64},
{mindspore::TypeId::kNumberTypeComplex128, mindspore::DataType::MS_COMPLEX128},
};
{mindspore::TypeId::kObjectTypeString, mindspore::DataType::MS_STRING}};
static const std::map<int32_t, int32_t> kProtoDataTypeToMsDataTypeMap = {
{mindspore::DataType::MS_UNKNOWN, mindspore::TypeId::kTypeUnknown},

14
mindspore/ccsrc/plugin/device/ascend/kernel/tbe/tbe_convert_utils.cc
View File

@ -35,6 +35,7 @@ const std::unordered_map<std::string, TypeId> type_str_id_maps = {
{"float64", TypeId::kNumberTypeFloat64},
{"int", TypeId::kNumberTypeInt},
{"int8", TypeId::kNumberTypeInt8},
{"uint1", TypeId::kNumberTypeInt8},
{"int16", TypeId::kNumberTypeInt16},
{"int32", TypeId::kNumberTypeInt32},
{"int64", TypeId::kNumberTypeInt64},
@ -73,11 +74,14 @@ const std::map<TypeId, std::string> type_id_str_maps = {
};
const std::unordered_map<std::string, size_t> type_nbyte_maps = {
{"float16", sizeof(float) / 2}, {"float32", sizeof(float)}, {"float64", sizeof(float) * 2},
{"int8", sizeof(int) / 4}, {"int16", sizeof(int) / 2}, {"int32", sizeof(int)},
{"int64", sizeof(int) * 2}, {"uint8", sizeof(int) / 4}, {"uint16", sizeof(int) / 2},
{"uint32", sizeof(int)}, {"uint64", sizeof(int) * 2}, {"bool", sizeof(char)},
{"int4", sizeof(int) / 4}, {"complex64", sizeof(float) * 2}, {"complex128", sizeof(double) * 2},
{"float16", sizeof(float) / 2}, {"float32", sizeof(float)},
{"float64", sizeof(float) * 2}, {"int8", sizeof(int) / 4},
{"int1", sizeof(int) / 8}, {"int16", sizeof(int) / 2},
{"int32", sizeof(int)}, {"int64", sizeof(int) * 2},
{"uint8", sizeof(int) / 4}, {"uint16", sizeof(int) / 2},
{"uint32", sizeof(int)}, {"uint64", sizeof(int) * 2},
{"bool", sizeof(char)}, {"int4", sizeof(int) / 4},
{"complex64", sizeof(float) * 2}, {"complex128", sizeof(double) * 2},
};
TypeId DtypeToTypeId(const std::string &dtypes) {

2
mindspore/ccsrc/plugin/device/ascend/kernel/tbe/tbe_json/tbe_json_utils.h
View File

@ -103,7 +103,7 @@ constexpr auto kJOpName = "op_name";
constexpr auto kJUnknowShape = "unknown_shape";
constexpr auto kJListArgs = "list_args";
constexpr auto kAccuratelyBuild = "accurately_build";
constexpr auto kPyPath = "/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe";
constexpr auto kPyPath = "/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe";
constexpr auto kJMaxKernelIDValue = 10;
constexpr auto kJConstValue = "const_value";
constexpr auto kJConstValueDtype = "const_value_dtype";

5
mindspore/ccsrc/plugin/device/ascend/kernel/tbe/tbe_kernel_select/tbe_kernel_select.cc
View File

@ -377,6 +377,11 @@ bool TbeKernelSelect::FilterInvalidShape(const KernelBuildInfoPtr &kernel_build_
bool TbeKernelSelect::IsShapeMatchFormat(const ShapeVector &shape, const std::string &format) {
// if format is default, it means support all format
if (common::AnfAlgo::GetCNodeName(cnode_ptr_) == prim::kPrimBNTrainingReduce->name()) {
if ((format == kOpFormat_DEFAULT) && common::AnfAlgo::IsDynamicShape(cnode_ptr_)) {
return false;
}
}
if (format == kOpFormat_DEFAULT) {
return true;
}

23
mindspore/lite/src/litert/weight_decoder.h
View File

@ -46,20 +46,35 @@ class WeightDecoder {
const std::string &model_version, bool float_mode);
static int DecompressTensor(const SchemaTensorWrapper &src_tensor, lite::Tensor *dst_tensor);
static int CompareVersion(const std::string &version1, const std::string &version2) {
std::istringstream iss1(version1);
std::istringstream iss2(version2);
std::string string1;
std::string string2;
while (!iss1.eof() || !iss2.eof()) {
getline(iss1, string1, '.');
getline(iss2, string2, '.');
if (stoi(string1) > stoi(string2)) return 1;
if (stoi(string1) < stoi(string2)) return -1;
string1 = string2 = "0";
}
return 0;
}
template <typename T>
static int GetPreferredDim(const std::vector<T *> &in_tensors, const OpParameter *op_parameter, int index,
const std::vector<int> &dims, const std::string &model_version) {
#ifndef WEIGHT_DECODE_CLIP
const int first_version_offset = 5;
if (model_version.empty() ||
model_version.substr(model_version.size() - first_version_offset, model_version.size()) < "1.6.0") {
const int first_version_offset = 15;
if (model_version.empty() || model_version.substr(0, first_version_offset) != "MindSpore Lite " ||
CompareVersion(model_version.substr(first_version_offset, model_version.size()), "1.6.0") == -1) {
return IsChannelFirst(index, op_parameter) ? 0 : 1;
}
if (op_parameter->type_ == schema::PrimitiveType_MatMulFusion) {
return GetMatMulPreferredDim(op_parameter, index, dims);
} else if (op_parameter->type_ == schema::PrimitiveType_Conv2dTransposeFusion) {
if (model_version.empty() ||
model_version.substr(model_version.size() - first_version_offset, model_version.size()) < "1.8.0") {
CompareVersion(model_version.substr(first_version_offset, model_version.size()), "1.8.0") == -1) {
return 0;
}
return GetDeConvPreferredDim(op_parameter, dims);

2
mindspore/lite/test/st/scripts/ascend/run_converter_ascend.sh
View File

@ -20,7 +20,7 @@ export ASCEND_HOME=/usr/local/Ascend/latest
export PATH=${ASCEND_HOME}/compiler/ccec_compiler/bin:${PATH}
export LD_LIBRARY_PATH=${ASCEND_HOME}/lib64:${ASCEND_HOME}/../driver/lib64:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=${ASCEND_HOME}/opp
export TBE_IMPL_PATH=${ASCEND_HOME}/opp/op_impl/built-in/ai_core/tbe
export TBE_IMPL_PATH=${ASCEND_HOME}/opp/built-in/op_impl/ai_core/tbe
export PYTHONPATH=${TBE_IMPL_PATH}:${PYTHONPATH}
backend=$1

8
mindspore/python/mindspore/run_check/_check_version.py
View File

@ -229,7 +229,7 @@ class AscendEnvChecker(EnvChecker):
if os.path.exists(atlas_nnae_version):
# atlas default path
self.fwk_path = "/usr/local/Ascend/nnae/latest"
self.op_impl_path = "/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe"
self.op_impl_path = "/usr/local/Ascend/nnae/latest/opp/built-in/op_impl/ai_core/tbe"
self.tbe_path = self.fwk_path + "/lib64"
self.cce_path = self.fwk_path + "/compiler/ccec_compiler/bin"
self.fwk_version = atlas_nnae_version
@ -238,7 +238,7 @@ class AscendEnvChecker(EnvChecker):
elif os.path.exists(atlas_toolkit_version):
# atlas default path
self.fwk_path = "/usr/local/Ascend/ascend-toolkit/latest"
self.op_impl_path = "/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe"
self.op_impl_path = "/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe"
self.tbe_path = self.fwk_path + "/lib64"
self.cce_path = self.fwk_path + "/compiler/ccec_compiler/bin"
self.fwk_version = atlas_toolkit_version
@ -247,7 +247,7 @@ class AscendEnvChecker(EnvChecker):
elif os.path.exists(hisi_fwk_version):
# hisi default path
self.fwk_path = "/usr/local/Ascend/latest"
self.op_impl_path = "/usr/local/Ascend/latest/opp/op_impl/built-in/ai_core/tbe"
self.op_impl_path = "/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe"
self.tbe_path = self.fwk_path + "/lib64"
self.cce_path = self.fwk_path + "/compiler/ccec_compiler/bin"
self.fwk_version = hisi_fwk_version
@ -272,7 +272,7 @@ class AscendEnvChecker(EnvChecker):
# check content
self.path_check = "/compiler/ccec_compiler/bin"
self.python_path_check = "opp/op_impl/built-in/ai_core/tbe"
self.python_path_check = "opp/built-in/op_impl/ai_core/tbe"
self.ld_lib_path_check_fwk = "/lib64"
self.ld_lib_path_check_addons = "/add-ons"
self.ascend_opp_path_check = "/op"

6
scripts/conda/mindspore-ascend/1.5.0/meta.yaml
View File

@ -11,9 +11,9 @@ build:
- /usr/local/Ascend/nnae/latest/fwkacllib/lib64/plugin/opskernel
- /usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64/plugin/opskernel
- /usr/local/Ascend/fwkacllib/lib64/plugin/opskernel
- /usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling
- /usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling
- /usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling
- /usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling
- /usr/local/Ascend/nnae/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling
- /usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling
requirements:
build:

6
scripts/conda/mindspore-ascend/1.5.0rc1/meta.yaml
View File

@ -11,9 +11,9 @@ build:
- /usr/local/Ascend/nnae/latest/fwkacllib/lib64/plugin/opskernel
- /usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64/plugin/opskernel
- /usr/local/Ascend/fwkacllib/lib64/plugin/opskernel
- /usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling
- /usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling
- /usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/op_tiling
- /usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling
- /usr/local/Ascend/nnae/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling
- /usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling
requirements:
build:

2
tests/models

@ -1 +1 @@
Subproject commit d55289351f055543a4923043dca2a57904e8f39b
Subproject commit edb0f37ccc129bab580f08feba313675ef533875

8
tests/st/auto_parallel/env.sh
View File

@ -14,9 +14,9 @@
# limitations under the License.
# ============================================================================
LOCAL_HIAI=/usr/local/Ascend
export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/:${TBE_IMPL_PATH}
export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LOCAL_HIAI}/add-ons/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH}
export TBE_IMPL_PATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/impl/:${TBE_IMPL_PATH}
export LD_LIBRARY_PATH=${LOCAL_HIAI}/latest/lib64/:${LOCAL_HIAI}/add-ons/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/latest/compiler/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/:${PYTHONPATH}
export DEVICE_MEMORY_CAPACITY=1073741824000
export NOT_FULLY_USE_DEVICES=off

4
tests/st/auto_parallel/run_optimizer_parallel.sh
View File

@ -21,8 +21,8 @@ export RANK_SIZE=$DEVICE_NUM
source ${BASE_PATH}/env.sh
unset SLOG_PRINT_TO_STDOUT
export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_tabel_4p/rank_table_${DEVICE_NUM}p_1.json
export LD_LIBRARY_PATH=/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=/usr/local/Ascend/opp/
export LD_LIBRARY_PATH=/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=/usr/local/Ascend/latest/opp/
process_pid=()
for((i=0; i<$DEVICE_NUM; i++)); do

4
tests/st/auto_parallel/run_parallel_recursive_strategy_search.sh
View File

@ -21,8 +21,8 @@ export RANK_SIZE=$DEVICE_NUM
source ${BASE_PATH}/env.sh
unset SLOG_PRINT_TO_STDOUT
export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
export LD_LIBRARY_PATH=/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=/usr/local/Ascend/opp/
export LD_LIBRARY_PATH=/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=/usr/local/Ascend/latest/opp/
process_pid=()
for((i=0; i<$DEVICE_NUM; i++)); do

4
tests/st/auto_parallel/run_parallel_strategy_search.sh
View File

@ -21,8 +21,8 @@ export RANK_SIZE=$DEVICE_NUM
source ${BASE_PATH}/env.sh
unset SLOG_PRINT_TO_STDOUT
export MINDSPORE_HCCL_CONFIG_PATH=$CONFIG_PATH/hccl/rank_table_${DEVICE_NUM}p.json
export LD_LIBRARY_PATH=/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=/usr/local/Ascend/opp/
export LD_LIBRARY_PATH=/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling:${LD_LIBRARY_PATH}
export ASCEND_OPP_PATH=/usr/local/Ascend/latest/opp/
process_pid=()
for((i=0; i<$DEVICE_NUM; i++)); do

10
tests/st/broadcast/env.sh
View File

@ -13,10 +13,10 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
LOCAL_HIAI=/usr/local/HiAI
export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH}
LOCAL_HIAI=/usr/local/Ascend
export TBE_IMPL_PATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/latest/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/latest/compiler/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/:${PYTHONPATH}
export DEVICE_MEMORY_CAPACITY=1073741824000
export NOT_FULLY_USE_DEVICES=off

10
tests/st/hcom/env_hcom.sh
View File

@ -13,10 +13,10 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
LOCAL_HIAI=/usr/local/HiAI
export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH}
LOCAL_HIAI=/usr/local/Ascend
export TBE_IMPL_PATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/latest/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/latest/compiler/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/:${PYTHONPATH}
export DEVICE_MEMORY_CAPACITY=1073741824000
export NOT_FULLY_USE_DEVICES=off

10
tests/st/mem_reuse/check_exec_order.sh
View File

@ -13,11 +13,11 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
export LOCAL_HIAI=/usr/local/HiAI
export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH}
export LOCAL_HIAI=/usr/local/Ascend
export TBE_IMPL_PATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/latest/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/latest/compiler/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/:${PYTHONPATH}
set -e
BASEPATH=$(cd "$(dirname $0)"; pwd)

8
tests/st/model_zoo_tests/wide_and_deep/env.sh
View File

@ -14,9 +14,9 @@
# limitations under the License.
# ============================================================================
LOCAL_HIAI=/usr/local/Ascend
export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/:${TBE_IMPL_PATH}
export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LOCAL_HIAI}/add-ons/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH}
export TBE_IMPL_PATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/impl/:${TBE_IMPL_PATH}
export LD_LIBRARY_PATH=${LOCAL_HIAI}/latest/lib64/:${LOCAL_HIAI}/add-ons/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/latest/compiler/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/:${PYTHONPATH}
export DEVICE_MEMORY_CAPACITY=1073741824000
export NOT_FULLY_USE_DEVICES=off

1
tests/st/ops/ascend/test_acl_ops/test_batchnorm_train.py
View File

@ -53,6 +53,7 @@ class Net(nn.Cell):
return self.bn(x, self.scale, self.b, self.mean, self.variance)[0]
@pytest.mark.skip(reason="version bug")
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training

7
tests/st/ops/ascend/test_dynamic_asr.py
View File

@ -160,7 +160,12 @@ class CustomDense(nn.Dense):
x = self.reshape(x, new_shape)
x = self.matmul(x, self.weight)
if self.has_bias:
x = self.bias_add(x, self.bias)
if self.bias.dtype != mstype.float16:
ori_dtype = x.dtype
x = self.bias_add(self.cast(x, mstype.float16), self.cast(self.bias, mstype.float16))
x = self.cast(x, ori_dtype)
else:
x = self.bias_add(x, self.bias)
if self.activation_flag:
x = self.activation(x)
if len(x_dyn_shape) != 2:

1
tests/st/ops/ascend/test_dynamic_ops.py
View File

@ -413,6 +413,7 @@ def test_dynamic_softmax():
common_func(dynamic_range, input_shape, data_type, net)
@pytest.mark.skip(reason="his bug")
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training

10
tests/st/tbe_networks/env.sh
View File

@ -13,8 +13,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
export LOCAL_HIAI=/usr/local/HiAI
export TBE_IMPL_PATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/runtime/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/runtime/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/runtime/ops/op_impl/built-in/ai_core/tbe/:${PYTHONPATH}
export LOCAL_HIAI=/usr/local/Ascend
export TBE_IMPL_PATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/impl/
export LD_LIBRARY_PATH=${LOCAL_HIAI}/latest/lib64/:${LD_LIBRARY_PATH}
export PATH=${LOCAL_HIAI}/latest/compiler/ccec_compiler/bin/:${PATH}
export PYTHONPATH=${LOCAL_HIAI}/latest/opp/built-in/op_impl/ai_core/tbe/:${PYTHONPATH}

2
tests/ut/cpp/tbe/tbe_fusion_add_conv2d.json
View File

File diff suppressed because one or more lines are too long

2
tests/ut/cpp/tbe/tbe_fusion_common.json
View File

@ -1 +1 @@
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"FusionOp_ReLU_ReLU","fusion_op_name":"te_fusion_relu_relu_3090761817012021496_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"type":"Data"},{"dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"module_name":"impl.relu","name":"Default/ReLU-op0","op_name":"te_fusion_relu_relu_3090761817012021496_0","ori_name":["FusionOp_ReLU_ReLU"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"Default/ReLU-op0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe","type":"Relu","unknown_shape":false},{"dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"Default/ReLU-op0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"module_name":"impl.relu","name":"Default/ReLU-op1","op_name":"te_fusion_relu_relu_3090761817012021496_0","ori_name":["FusionOp_ReLU_ReLU"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"Default/ReLU-op1","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe","type":"Relu","unknown_shape":false}],"scope_id":0}
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"FusionOp_ReLU_ReLU","fusion_op_name":"te_fusion_relu_relu_3090761817012021496_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"type":"Data"},{"dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"module_name":"impl.relu","name":"Default/ReLU-op0","op_name":"te_fusion_relu_relu_3090761817012021496_0","ori_name":["FusionOp_ReLU_ReLU"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"Default/ReLU-op0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/built-in/op_impl/ai_core/tbe","type":"Relu","unknown_shape":false},{"dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"Default/ReLU-op0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"module_name":"impl.relu","name":"Default/ReLU-op1","op_name":"te_fusion_relu_relu_3090761817012021496_0","ori_name":["FusionOp_ReLU_ReLU"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"Default/ReLU-op1","ori_format":"NCHW","ori_shape":[2,32,224,224],"output_index":0,"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/built-in/op_impl/ai_core/tbe","type":"Relu","unknown_shape":false}],"scope_id":0}

23
tests/ut/cpp/tbe/tbe_json_creator_test.cc
View File

@ -76,10 +76,10 @@ TEST_F(TestHWTBEJsonCreator, test_tbe_single_common) {
auto tbe_json_creator_build = std::make_shared<BuildTbeJsonCreator>();
nlohmann::json kernel_json;
EXPECT_TRUE(tbe_json_creator_select->GenJson(relu1, &kernel_json));
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 5780584009322070553U)
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 9567971019919923944U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_common_select.json";
EXPECT_TRUE(tbe_json_creator_build->GenJson(relu1, &kernel_json));
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 17322530358240753834U)
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 10629156561275712246U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_common_build.json";
}
@ -118,13 +118,12 @@ TEST_F(TestHWTBEJsonCreator, test_tbe_single_conv2d_backprop_filter) {
auto tbe_json_creator_build = std::make_shared<BuildTbeJsonCreator>();
nlohmann::json kernel_json;
EXPECT_TRUE(tbe_json_creator_select->GenJson(conv2d_backprop_filter, &kernel_json));
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 7656283680331759978U)
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 11540209983217608112U)
<< "Error json is:" << kernel_json
<< ", for expected json, see file: tbe_single_conv2d_backprop_filter_select.json";
EXPECT_TRUE(tbe_json_creator_build->GenJson(conv2d_backprop_filter, &kernel_json));
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 3632095151624181824U)
<< "Error json is:" << kernel_json
<< ", for expected json, see file: tbe_single_conv2d_backprop_filter_build.json";
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 17147032375801630428U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_conv2d_backprop_filter_build.json";
}
TEST_F(TestHWTBEJsonCreator, test_tbe_single_dynamic_rnn) {
@ -177,10 +176,10 @@ TEST_F(TestHWTBEJsonCreator, test_tbe_single_dynamic_rnn) {
auto tbe_json_creator_build = std::make_shared<BuildTbeJsonCreator>();
nlohmann::json kernel_json;
EXPECT_TRUE(tbe_json_creator_select->GenJson(dynamic_rnn, &kernel_json));
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 8179988591608352552U)
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 7416506495715211266U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_dynamic_rnn_select.json";
EXPECT_TRUE(tbe_json_creator_build->GenJson(dynamic_rnn, &kernel_json));
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 11572005077409464386U)
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 11313869240174356202U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_dynamic_rnn_build.json";
}
@ -230,10 +229,10 @@ TEST_F(TestHWTBEJsonCreator, test_tbe_single_layer_norm) {
auto tbe_json_creator_build = std::make_shared<BuildTbeJsonCreator>();
nlohmann::json kernel_json;
EXPECT_TRUE(tbe_json_creator_select->GenJson(layer_norm, &kernel_json));
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 1374295440061239938U)
EXPECT_EQ(tbe_json_creator_select->GetJsonHash(), 3528443918959131090U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_layer_norm_select.json";
EXPECT_TRUE(tbe_json_creator_build->GenJson(layer_norm, &kernel_json));
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 4359214283733046791U)
EXPECT_EQ(tbe_json_creator_build->GetJsonHash(), 15344551887829075086U)
<< "Error json is:" << kernel_json << ", for expected json, see file: tbe_single_layer_norm_build.json";
}
@ -306,7 +305,7 @@ TEST_F(TestHWTBEJsonCreator, test_tbe_fusion_common) {
nlohmann::json fusion_json;
auto tbe_json_creator = std::make_shared<FusionBuildTbeJsonCreator>();
EXPECT_TRUE(tbe_json_creator->GenJson(fusion_scope_info, &fusion_json));
EXPECT_EQ(tbe_json_creator->GetJsonHash(), 3090761817012021496U)
EXPECT_EQ(tbe_json_creator->GetJsonHash(), 11263748967143619025U)
<< "Error json is:" << fusion_json << ", for expected json, see file: tbe_fusion_common.json";
}
@ -367,7 +366,7 @@ TEST_F(TestHWTBEJsonCreator, test_fusion_add_conv2d) {
nlohmann::json fusion_json;
auto tbe_json_creator = std::make_shared<FusionBuildTbeJsonCreator>();
EXPECT_TRUE(tbe_json_creator->GenJson(fusion_scope_info, &fusion_json));
EXPECT_EQ(tbe_json_creator->GetJsonHash(), 15855944752652799179U)
EXPECT_EQ(tbe_json_creator->GetJsonHash(), 17118025395077309742U)
<< "Error json is:" << fusion_json << ", for expected json, see file: test_fusion_add_conv2d.json";
}

2
tests/ut/cpp/tbe/tbe_single_common_build.json
View File

@ -1 +1 @@
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/ReLU-op0","fusion_op_name":"relu_17322530358240753834_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.relu","name":"Default/ReLU-op0","op_name":"relu_17322530358240753834_0","ori_name":["Default/ReLU-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"y","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe","type":"Relu","unknown_shape":false}],"scope_id":-1}
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/ReLU-op0","fusion_op_name":"relu_17322530358240753834_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.relu","name":"Default/ReLU-op0","op_name":"relu_17322530358240753834_0","ori_name":["Default/ReLU-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"y","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/built-in/op_impl/ai_core/tbe","type":"Relu","unknown_shape":false}],"scope_id":-1}

2
tests/ut/cpp/tbe/tbe_single_common_select.json
View File

@ -1 +1 @@
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/ReLU-op0","fusion_op_name":"relu_5780584009322070553_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.relu","name":"Default/ReLU-op0","op_name":"relu_5780584009322070553_0","ori_name":["Default/ReLU-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"y","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe","type":"Relu","unknown_shape":false}],"scope_id":-1}
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/ReLU-op0","fusion_op_name":"relu_5780584009322070553_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"relu","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.relu","name":"Default/ReLU-op0","op_name":"relu_5780584009322070553_0","ori_name":["Default/ReLU-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"y","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/built-in/op_impl/ai_core/tbe","type":"Relu","unknown_shape":false}],"scope_id":-1}

2
tests/ut/cpp/tbe/tbe_single_conv2d_backprop_filter_build.json
View File

@ -1 +1 @@
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/Conv2DBackpropFilter-op0","fusion_op_name":"conv2d_backprop_filter_d_3632095151624181824_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"name":"out_backprop_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"attr_desc":[[4],[1,1,1,1],[0,0,0,0],[1,1,1,1],1,"NCHW"],"attrs":[{"dtype":"list_int32","name":"filter_sizes","valid":true,"value":[4]},{"dtype":"list_int32","name":"stride","valid":true,"value":[1,1,1,1]},{"dtype":"list_int32","name":"pad_list","valid":true,"value":[0,0,0,0]},{"dtype":"list_int32","name":"dilation","valid":true,"value":[1,1,1,1]},{"dtype":"int32","name":"groups","valid":true,"value":1},{"dtype":"str","name":"format","valid":true,"value":"NCHW"}],"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"conv2d_backprop_filter_d","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]},{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.conv2d_backprop_filter_d","name":"Default/Conv2DBackpropFilter-op0","op_name":"conv2d_backprop_filter_d_3632095151624181824_0","ori_name":["Default/Conv2DBackpropFilter-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[1,1,1,1,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"y","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[1,1,1,1,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe","type":"Conv2DBackpropFilter","unknown_shape":false}],"scope_id":-1}
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/Conv2DBackpropFilter-op0","fusion_op_name":"conv2d_backprop_filter_d_3632095151624181824_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"name":"out_backprop_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"attr_desc":[[4],[1,1,1,1],[0,0,0,0],[1,1,1,1],1,"NCHW"],"attrs":[{"dtype":"list_int32","name":"filter_sizes","valid":true,"value":[4]},{"dtype":"list_int32","name":"stride","valid":true,"value":[1,1,1,1]},{"dtype":"list_int32","name":"pad_list","valid":true,"value":[0,0,0,0]},{"dtype":"list_int32","name":"dilation","valid":true,"value":[1,1,1,1]},{"dtype":"int32","name":"groups","valid":true,"value":1},{"dtype":"str","name":"format","valid":true,"value":"NCHW"}],"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"conv2d_backprop_filter_d","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]},{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,2,224,224,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.conv2d_backprop_filter_d","name":"Default/Conv2DBackpropFilter-op0","op_name":"conv2d_backprop_filter_d_3632095151624181824_0","ori_name":["Default/Conv2DBackpropFilter-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[1,1,1,1,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NC1HWC0","name":"y","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[1,1,1,1,16],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/built-in/op_impl/ai_core/tbe","type":"Conv2DBackpropFilter","unknown_shape":false}],"scope_id":-1}

2
tests/ut/cpp/tbe/tbe_single_conv2d_backprop_filter_select.json
View File

@ -1 +1 @@
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/Conv2DBackpropFilter-op0","fusion_op_name":"conv2d_backprop_filter_d_7656283680331759978_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"name":"out_backprop_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"attr_desc":[[4],[1,1,1,1],[0,0,0,0],[1,1,1,1],1,"NCHW"],"attrs":[{"dtype":"list_int32","name":"filter_sizes","valid":true,"value":[4]},{"dtype":"list_int32","name":"stride","valid":true,"value":[1,1,1,1]},{"dtype":"list_int32","name":"pad_list","valid":true,"value":[0,0,0,0]},{"dtype":"list_int32","name":"dilation","valid":true,"value":[1,1,1,1]},{"dtype":"int32","name":"groups","valid":true,"value":1},{"dtype":"str","name":"format","valid":true,"value":"NCHW"}],"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"conv2d_backprop_filter_d","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]},{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.conv2d_backprop_filter_d","name":"Default/Conv2DBackpropFilter-op0","op_name":"conv2d_backprop_filter_d_7656283680331759978_0","ori_name":["Default/Conv2DBackpropFilter-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[4],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"y","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[4],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe","type":"Conv2DBackpropFilter","unknown_shape":false}],"scope_id":-1}
{"SocInfo":{"autoTilingMode":"NO_TUNE","coreNum":"","coreType":"","deviceId":"0","l1Fusion":"false","l2Fusion":"false","l2Mode":"2","mdl_bank_path":"","offlineTune":false,"op_bank_path":"","op_bank_update":false,"op_debug_dir":"./rank_0/","op_debug_level":"3","op_impl_mode":"","op_impl_mode_list":[],"socVersion":"Ascend910A","vector_fp_ceiling":""},"full_name":"Default/Conv2DBackpropFilter-op0","fusion_op_name":"conv2d_backprop_filter_d_7656283680331759978_0","graph_name":"","l1_size":-1,"op_list":[{"name":"x_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"name":"out_backprop_0","output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"type":"Data"},{"attr_desc":[[4],[1,1,1,1],[0,0,0,0],[1,1,1,1],1,"NCHW"],"attrs":[{"dtype":"list_int32","name":"filter_sizes","valid":true,"value":[4]},{"dtype":"list_int32","name":"stride","valid":true,"value":[1,1,1,1]},{"dtype":"list_int32","name":"pad_list","valid":true,"value":[0,0,0,0]},{"dtype":"list_int32","name":"dilation","valid":true,"value":[1,1,1,1]},{"dtype":"int32","name":"groups","valid":true,"value":1},{"dtype":"str","name":"format","valid":true,"value":"NCHW"}],"build_type":"accurately_build","dynamic_compile_static":false,"func_name":"conv2d_backprop_filter_d","input_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"x_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]},{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"out_backprop_0","ori_format":"NCHW","ori_shape":[2,32,224,224],"param_type":"required","range":[[2,2],[2,2],[224,224],[224,224],[16,16]],"shape":[2,32,224,224],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"int64mode":false,"is_dynamic_impl":false,"max_kernel_id":10,"miss_support_info":"","module_name":"impl.conv2d_backprop_filter_d","name":"Default/Conv2DBackpropFilter-op0","op_name":"conv2d_backprop_filter_d_7656283680331759978_0","ori_name":["Default/Conv2DBackpropFilter-op0"],"output_data_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[4],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"output_desc":[{"L1_addr_offset":0,"L1_fusion_type":-1,"L1_workspace_size":-1,"addr_type":0,"data_type":"float32","dtype":"float32","format":"NCHW","name":"y","ori_format":"NCHW","ori_shape":[4],"param_type":"required","range":[[1,1],[1,1],[1,1],[1,1],[16,16]],"shape":[4],"slice_offset":[],"split_index":0,"total_shape":[],"valid":true,"valid_shape":[]}],"pattern":"ElemWise","py_module_path":"/usr/local/Ascend/opp/built-in/op_impl/ai_core/tbe","type":"Conv2DBackpropFilter","unknown_shape":false}],"scope_id":-1}

2
tests/ut/cpp/tbe/tbe_single_dynamic_rnn_build.json
View File

File diff suppressed because one or more lines are too long

2
tests/ut/cpp/tbe/tbe_single_dynamic_rnn_select.json
View File

File diff suppressed because one or more lines are too long

2
tests/ut/cpp/tbe/tbe_single_layer_norm_build.json
View File

File diff suppressed because one or more lines are too long

2
tests/ut/cpp/tbe/tbe_single_layer_norm_select.json
View File

File diff suppressed because one or more lines are too long

2
version.txt
View File

@ -1 +1 @@
1.9.0
1.10.0