Paddle/python/paddle/fluid/contrib/tests/CMakeLists.txt

file(GLOB TEST_OPS RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}" "test_*.py")
string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")

if(APPLE OR WIN32 OR NOT WITH_MKL)
    list(REMOVE_ITEM TEST_OPS test_calibration_resnet50)
    list(REMOVE_ITEM TEST_OPS test_calibration_mobilenetv1)
endif()

foreach(src ${TEST_OPS})
    if(src MATCHES "test_calibration_*")
        py_test(${src} SRCS ${src}.py ENVS FLAGS_use_mkldnn=true FLAGS_OMP_NUM_THREADS=${CPU_NUM_THREADS_ON_CI})
    else()
        py_test(${src} SRCS ${src}.py)
    endif()
endforeach()