fix bug of test for cinn with pir (PaddlePaddle#58775)

danleifeng · Nov 8, 2023 · 1255d56 · 1255d56
1 parent 2b3d899
commit 1255d56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/ir/pir/cinn/CMakeLists.txt b/test/ir/pir/cinn/CMakeLists.txt
@@ -11,7 +11,7 @@ if(WITH_GPU)
       COMMAND
         ${CMAKE_COMMAND} -E env
         PYTHONPATH=${CMAKE_BINARY_DIR}:${CMAKE_BINARY_DIR}/python/:$ENV{PYTHONPATH}
-        FLAGS_enable_pir_api=1 python3.10
+        FLAGS_enable_pir_api=1 ${PYTHON_EXECUTABLE}
         ${CMAKE_CURRENT_SOURCE_DIR}/${cinn_pir_test_name}.py
       WORKING_DIRECTORY ${CMAKE_BINARY_DIR})
     set_tests_properties(${cinn_pir_test_name} PROPERTIES LABELS