diff --git a/test/prim/model/CMakeLists.txt b/test/prim/model/CMakeLists.txt index 1395be8b3a243e06570bf34d08e0097616ce8b17..c37a25924aa970fb8e3c0427ffd757bc892706a4 100644 --- a/test/prim/model/CMakeLists.txt +++ b/test/prim/model/CMakeLists.txt @@ -8,19 +8,19 @@ foreach(TEST_OP ${TEST_OPS}) py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS}) endforeach() -#set_tests_properties(test_resnet_prim PROPERTIES TIMEOUT 850) +set_tests_properties(test_resnet_prim PROPERTIES TIMEOUT 850) set_tests_properties(test_bert_prim PROPERTIES TIMEOUT 500) set_tests_properties(test_prim_simplenet_cinn PROPERTIES TIMEOUT 120) if(WITH_CINN) - #set_tests_properties(test_resnet_cinn PROPERTIES TIMEOUT 850) - #set_tests_properties(test_resnet_prim_cinn PROPERTIES TIMEOUT 850) + set_tests_properties(test_resnet_cinn PROPERTIES TIMEOUT 850) + set_tests_properties(test_resnet_prim_cinn PROPERTIES TIMEOUT 850) set_tests_properties(test_bert_cinn PROPERTIES TIMEOUT 500) set_tests_properties(test_bert_prim_cinn PROPERTIES TIMEOUT 500) - #set_tests_properties(test_resnet_prim PROPERTIES LABELS "RUN_TYPE=CINN") - #set_tests_properties(test_resnet_cinn PROPERTIES LABELS "RUN_TYPE=CINN") - #set_tests_properties(test_resnet_prim_cinn PROPERTIES LABELS "RUN_TYPE=CINN") + set_tests_properties(test_resnet_prim PROPERTIES LABELS "RUN_TYPE=CINN") + set_tests_properties(test_resnet_cinn PROPERTIES LABELS "RUN_TYPE=CINN") + set_tests_properties(test_resnet_prim_cinn PROPERTIES LABELS "RUN_TYPE=CINN") set_tests_properties(test_bert_prim PROPERTIES LABELS "RUN_TYPE=CINN") set_tests_properties(test_bert_cinn PROPERTIES LABELS "RUN_TYPE=CINN") set_tests_properties(test_bert_prim_cinn PROPERTIES LABELS "RUN_TYPE=CINN") diff --git a/test/prim/model/test_resnet_cinn.py b/test/prim/model/test_resnet_cinn.py index 41636462dda9e4b6de80c00e5f9f6672b50497c1..b2d6004aa2f7b6de6c24a610b16c174d1413ee64 100644 --- a/test/prim/model/test_resnet_cinn.py +++ b/test/prim/model/test_resnet_cinn.py @@ -45,16 +45,16 @@ epoch_num = 1 # note: Version 2.0 momentum is fused to OP when L2Decay is available, and the results are different from the fluid version. # The results in ci as as follows: DY2ST_CINN_GT = [ - 5.828789710998535, - 8.340764999389648, - 4.998946666717529, - 8.4613676071167, - 8.033733367919922, - 7.352842807769775, - 9.8336181640625, - 8.22379207611084, - 8.195695877075195, - 10.508796691894531, + 5.847333908081055, + 8.347576141357422, + 5.1415300369262695, + 8.373777389526367, + 8.05331802368164, + 7.437496185302734, + 9.630914688110352, + 8.547889709472656, + 8.343082427978516, + 10.203847885131836, ] if core.is_compiled_with_cuda(): diff --git a/test/prim/model/test_resnet_prim.py b/test/prim/model/test_resnet_prim.py index 6ede8ca3adb050b5fb4bc6124c10138a2946ad28..c423702a31db0757a9fdda196aa2c350ce2f9763 100644 --- a/test/prim/model/test_resnet_prim.py +++ b/test/prim/model/test_resnet_prim.py @@ -46,16 +46,16 @@ epoch_num = 1 # note: Version 2.0 momentum is fused to OP when L2Decay is available, and the results are different from the fluid version. # The results in ci as as follows: DY2ST_PRIM_GT = [ - 5.82879114151001, - 8.33370590209961, - 5.104889392852783, - 8.546337127685547, - 8.263965606689453, - 7.413934230804443, - 9.569124221801758, - 8.251557350158691, - 8.513609886169434, - 10.603094100952148, + 5.847333908081055, + 8.368712425231934, + 4.989010334014893, + 8.523179054260254, + 7.997398376464844, + 7.601831436157227, + 9.777579307556152, + 8.428393363952637, + 8.581992149353027, + 10.313587188720703, ] if core.is_compiled_with_cuda(): diff --git a/test/prim/model/test_resnet_prim_cinn.py b/test/prim/model/test_resnet_prim_cinn.py index 9804d4dd6b0b8c38915fc31f7aa803da830edae0..d3ad08e80c51e23745334ce8a3de119e20557515 100644 --- a/test/prim/model/test_resnet_prim_cinn.py +++ b/test/prim/model/test_resnet_prim_cinn.py @@ -46,17 +46,18 @@ epoch_num = 1 # note: Version 2.0 momentum is fused to OP when L2Decay is available, and the results are different from the fluid version. # The results in ci as as follows: DY2ST_PRIM_CINN_GT = [ - 5.828786849975586, - 8.332863807678223, - 5.041562080383301, - 8.514982223510742, - 7.9860992431640625, - 7.491837501525879, - 9.559739112854004, - 8.430597305297852, - 8.109201431274414, - 10.224763870239258, + 5.847333908081055, + 8.342670440673828, + 5.130363941192627, + 8.511886596679688, + 8.13458251953125, + 7.35969352722168, + 9.874241828918457, + 8.126291275024414, + 8.637175559997559, + 10.385666847229004, ] + if core.is_compiled_with_cuda(): paddle.set_flags({'FLAGS_cudnn_deterministic': True})