openharmony/ai_neural_network_runtime
1
0
Fork 0
mirror of https://gitee.com/openharmony/ai_neural_network_runtime synced 2024-12-12 10:15:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

codeCheckx修复

Browse Source 
Signed-off-by: 周翔 <zhouxiang78@h-partners.com>
This commit is contained in:
周翔 2024-06-05 11:15:02 +08:00
parent 0d5bcba8c7
commit 6e5fe13f68
5 changed files with 19 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/unittest/BUILD.gn
View File

@ -17,6 +17,6 @@ group("unittest") {
testonly = true
deps = [
"components:components_unittest",
"ops:ops_unittest"
"ops:ops_unittest",
]
}

3
test/unittest/components/v1_0/inner_model/inner_model_test.cpp
View File

@ -241,7 +241,8 @@ HWTEST_F(InnerModelTest, inner_model_build_from_lite_graph_001, TestSize.Level1)
}
}
EXPECT_EQ(OH_NN_INVALID_PARAMETER, m_innerModelTest.BuildFromLiteGraph(nullptr, buffer, modelName, isProfiling, opLayouts));
EXPECT_EQ(OH_NN_INVALID_PARAMETER, m_innerModelTest
.BuildFromLiteGraph(nullptr, buffer, modelName, isProfiling, opLayouts));
}
/**

16
test/unittest/components/v1_0/neural_network_runtime_test/neural_network_runtime_test.cpp
View File

@ -25,9 +25,7 @@
namespace OHOS {
namespace NeuralNetworkRuntime {
OH_NN_ReturnCode HDIDeviceV1_0::PrepareModel(std::shared_ptr<const mindspore::lite::LiteGraph> model,
const Buffer& quantBuffer,
const ModelConfig& config,
std::shared_ptr<PreparedModel>& preparedModel)
const Buffer& quantBuffer, const ModelConfig& config, std::shared_ptr<PreparedModel>& preparedModel)
{
if (model == nullptr) {
return OH_NN_INVALID_PARAMETER;
@ -224,16 +222,16 @@ void NeuralNetworkRuntimeTest::SetInputAndOutput(Executor& executor)
size_t input1Index = 0;
int32_t inputDims[2] = {3, 4};
size_t lengthSize = 12 * sizeof(float);
size_t * length = &lengthSize;
size_t *length = &lengthSize;
size_t minInputDims = 1;
size_t maxInputDims = 12;
size_t *minInputDimsAdress = &minInputDims;
size_t **minInputDimsAdressA = &minInputDimsAdress;
size_t * minInputDimsAdress = &minInputDims;
size_t ** minInputDimsAdressA = &minInputDimsAdress;
size_t * maxInputDimsAdress = &maxInputDims;
size_t ** maxInputDimsAdressA = &maxInputDimsAdress;
size_t *maxInputDimsAdress = &maxInputDims;
size_t **maxInputDimsAdressA = &maxInputDimsAdress;
m_tensor = {OH_NN_FLOAT32, 2, inputDims, nullptr, OH_NN_TENSOR};
EXPECT_EQ(OH_NN_SUCCESS, executor.GetInputDimRange(input1Index, minInputDimsAdressA, maxInputDimsAdressA, length));

3
test/unittest/components/v2_0/inner_model/inner_model_test.cpp
View File

@ -239,7 +239,8 @@ HWTEST_F(InnerModelTest, inner_model_build_from_lite_graph_001, TestSize.Level1)
}
}
EXPECT_EQ(OH_NN_INVALID_PARAMETER, m_innerModelTest.BuildFromLiteGraph(nullptr, buffer, modelName, isProfiling, opLayouts));
EXPECT_EQ(OH_NN_INVALID_PARAMETER, m_innerModelTest
.BuildFromLiteGraph(nullptr, buffer, modelName, isProfiling, opLayouts));
}
/**

16
test/unittest/components/v2_0/neural_network_runtime_test/neural_network_runtime_test.cpp
View File

@ -25,9 +25,7 @@
namespace OHOS {
namespace NeuralNetworkRuntime {
OH_NN_ReturnCode HDIDeviceV2_0::PrepareModel(std::shared_ptr<const mindspore::lite::LiteGraph> model,
const Buffer& quantBuffer,
const ModelConfig& config,
std::shared_ptr<PreparedModel>& preparedModel)
const Buffer& quantBuffer, const ModelConfig& config, std::shared_ptr<PreparedModel>& preparedModel)
{
if (model == nullptr) {
return OH_NN_INVALID_PARAMETER;
@ -224,16 +222,16 @@ void NeuralNetworkRuntimeTest::SetInputAndOutput(Executor& executor)
size_t input1Index = 0;
int32_t inputDims[2] = {3, 4};
size_t lengthSize = 12 * sizeof(float);
size_t * length = &lengthSize;
size_t *length = &lengthSize;
size_t minInputDims = 1;
size_t maxInputDims = 12;
size_t *minInputDimsAdress = &minInputDims;
size_t **minInputDimsAdressA = &minInputDimsAdress;
size_t * minInputDimsAdress = &minInputDims;
size_t ** minInputDimsAdressA = &minInputDimsAdress;
size_t * maxInputDimsAdress = &maxInputDims;
size_t ** maxInputDimsAdressA = &maxInputDimsAdress;
size_t *maxInputDimsAdress = &maxInputDims;
size_t **maxInputDimsAdressA = &maxInputDimsAdress;
m_tensor = {OH_NN_FLOAT32, 2, inputDims, nullptr, OH_NN_TENSOR};
EXPECT_EQ(OH_NN_SUCCESS, executor.GetInputDimRange(input1Index, minInputDimsAdressA, maxInputDimsAdressA, length));