From 5ab6ebaf9ccb0f11fe04be36d89f8322ecb4ccb5 Mon Sep 17 00:00:00 2001 From: Leo Chen Date: Fri, 24 Dec 2021 19:16:08 +0800 Subject: [PATCH] set env for test_standalone_executor (#38430) --- python/paddle/fluid/tests/unittests/interpreter/CMakeLists.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/interpreter/CMakeLists.txt b/python/paddle/fluid/tests/unittests/interpreter/CMakeLists.txt index 7692f8befdf..42d109c17ba 100644 --- a/python/paddle/fluid/tests/unittests/interpreter/CMakeLists.txt +++ b/python/paddle/fluid/tests/unittests/interpreter/CMakeLists.txt @@ -4,3 +4,5 @@ string(REPLACE ".py" "" TEST_INTERP_CASES "${TEST_INTERP_CASES}") foreach(target ${TEST_INTERP_CASES}) py_test_modules(${target} MODULES ${target}) endforeach() + +set_tests_properties(test_standalone_executor PROPERTIES ENVIRONMENT FLAGS_use_stream_safe_cuda_allocator=true) -- GitLab