mirror of
https://github.com/opencv/opencv.git
synced 2026-07-21 19:33:03 +04:00
Merge pull request #29449 from asmorkalov:as/skip_dnn_tests_32bit
Skip some BERT tests on 32-bit platforms as they do not fit into 2gb ram
This commit is contained in:
@@ -829,8 +829,9 @@ INSTANTIATE_TEST_CASE_P(/**/, Reproducibility_ViT_ONNX,
|
|||||||
typedef testing::TestWithParam<Target> Reproducibility_BERT_ONNX;
|
typedef testing::TestWithParam<Target> Reproducibility_BERT_ONNX;
|
||||||
TEST_P(Reproducibility_BERT_ONNX, Accuracy)
|
TEST_P(Reproducibility_BERT_ONNX, Accuracy)
|
||||||
{
|
{
|
||||||
|
applyTestTag(CV_TEST_TAG_MEMORY_2GB);
|
||||||
|
|
||||||
Target targetId = GetParam();
|
Target targetId = GetParam();
|
||||||
applyTestTag(targetId == DNN_TARGET_CPU ? CV_TEST_TAG_MEMORY_1GB : CV_TEST_TAG_MEMORY_2GB);
|
|
||||||
ASSERT_TRUE(ocl::useOpenCL() || targetId == DNN_TARGET_CPU || targetId == DNN_TARGET_CPU_FP16);
|
ASSERT_TRUE(ocl::useOpenCL() || targetId == DNN_TARGET_CPU || targetId == DNN_TARGET_CPU_FP16);
|
||||||
auto engine_forced = static_cast<EngineType>(
|
auto engine_forced = static_cast<EngineType>(
|
||||||
cv::utils::getConfigurationParameterSizeT("OPENCV_FORCE_DNN_ENGINE", ENGINE_AUTO));
|
cv::utils::getConfigurationParameterSizeT("OPENCV_FORCE_DNN_ENGINE", ENGINE_AUTO));
|
||||||
|
|||||||
Reference in New Issue
Block a user