diff --git a/examples/3_MultiGPU/CMakeLists.txt b/examples/3_MultiGPU/CMakeLists.txt index 464e37a5..99315d53 100644 --- a/examples/3_MultiGPU/CMakeLists.txt +++ b/examples/3_MultiGPU/CMakeLists.txt @@ -61,7 +61,7 @@ if (CMAKE_BUILD_TESTS) add_test( NAME multigpu_infer_fortran COMMAND - multigpu_infer_fortran ${PROJECT_BINARY_DIR}/saved_multigpu_model_cpu.pt + multigpu_infer_fortran ${PROJECT_BINARY_DIR}/saved_multigpu_model_cuda.pt # Command line argument: model file WORKING_DIRECTORY ${PROJECT_BINARY_DIR}) set_tests_properties(