RunAuxiliaryRestart.cmake

Go to the documentation of this file.

# Run from the test's private working directory, containing examples,
# param_file.petsc and the common JSON fixture. Each stage keeps its own files.
foreach(required EP_MPI_RUN EP_EXECUTABLE EP_COMPARATOR EP_RANKS)
  if(NOT DEFINED ${required} OR "${${required}}" STREQUAL "")
    message(FATAL_ERROR "RunAuxiliaryRestart requires ${required}")
  endif()
endforeach()
if(NOT EP_RANKS MATCHES "^[12]$")
  message(FATAL_ERROR "Auxiliary restart test supports one or two ranks")
endif()
if(NOT DEFINED EP_CONFIG)
  set(EP_CONFIG "ep_cook_auxiliary_logarithmic_stress.json")
endif()
if(NOT DEFINED EP_COMPARE_TOLERANCE)
  set(EP_COMPARE_TOLERANCE 1e-9)
endif()
get_filename_component(test_directory "." ABSOLUTE)
get_filename_component(config_file "${EP_CONFIG}" ABSOLUTE)
file(READ "${config_file}" configuration)
if(EP_RANKS EQUAL 1)
  string(JSON configuration SET "${configuration}" meshes 0 file_name
    "\"examples/cook_cantilever/cook.cub\"")
endif()
 
foreach(stage uninterrupted initial resumed missing_metadata wrong_formulation)
  set(stage_directory "${test_directory}/${stage}")
  file(MAKE_DIRECTORY "${stage_directory}")
  file(WRITE "${stage_directory}/input.json" "${configuration}\n")
  configure_file("${test_directory}/param_file.petsc"
    "${stage_directory}/param_file.petsc" COPYONLY)
  if(NOT EXISTS "${stage_directory}/examples")
    file(CREATE_LINK "${test_directory}/examples"
      "${stage_directory}/examples" SYMBOLIC RESULT link_result)
    if(NOT link_result STREQUAL "0")
      message(FATAL_ERROR "Cannot link ${stage} mesh fixtures: ${link_result}")
    endif()
  endif()
  file(REMOVE "${stage_directory}/final.vec"
    "${stage_directory}/final.vec.info")
endforeach()
 
function(run_auxiliary_restart_stage stage expected_error)
  set(stage_directory "${test_directory}/${stage}")
  execute_process(
    COMMAND "${EP_MPI_RUN}" ${EP_MPI_RUN_FLAGS} -np "${EP_RANKS}"
      "${EP_EXECUTABLE}" -json_config input.json ${EP_OPTIONS} ${ARGN}
    WORKING_DIRECTORY "${stage_directory}"
    RESULT_VARIABLE run_result
    OUTPUT_FILE "${stage_directory}/run.log"
    ERROR_FILE "${stage_directory}/run.log"
    TIMEOUT 180)
  file(READ "${stage_directory}/run.log" run_log)
  if(expected_error STREQUAL "")
    if(NOT run_result STREQUAL "0")
      message(FATAL_ERROR
        "Auxiliary ${stage} solve failed (${run_result}):\n${run_log}")
    endif()
    if(NOT EXISTS "${stage_directory}/final.vec")
      message(FATAL_ERROR "Auxiliary ${stage} solve did not save its final vector")
    endif()
  else()
    string(FIND "${run_log}" "${expected_error}" diagnostic_position)
    if(NOT run_result MATCHES "^[1-9][0-9]*$" OR
       diagnostic_position LESS 0)
      message(FATAL_ERROR
        "Expected ${stage} failure '${expected_error}', got '${run_result}':\n${run_log}")
    endif()
  endif()
endfunction()
 
set(final_vector_option -elastic_snes_view_solution binary:final.vec)
file(REMOVE "${test_directory}/initial/restart_1.dat"
  "${test_directory}/initial/restart_1.dat.layout.json")
run_auxiliary_restart_stage(uninterrupted ""
  -time 0 -write_restart 0
  -elastic_ts_dt 0.05 -elastic_ts_max_time 0.1 -elastic_ts_max_steps 2
  ${final_vector_option})
run_auxiliary_restart_stage(initial ""
  -time 0 -write_restart 1 -write_restart_every 1
  -elastic_ts_dt 0.05 -elastic_ts_max_time 0.05 -elastic_ts_max_steps 1
  ${final_vector_option})
 
set(checkpoint "${test_directory}/initial/restart_1.dat")
set(metadata "${checkpoint}.layout.json")
if(NOT EXISTS "${checkpoint}" OR NOT EXISTS "${metadata}")
  message(FATAL_ERROR "Initial auxiliary solve did not write a native restart pair")
endif()
file(READ "${metadata}" layout)
string(JSON formulation GET "${layout}" formulation)
if(NOT formulation STREQUAL "auxiliary_logarithmic_stress")
  message(FATAL_ERROR "Checkpoint does not identify the auxiliary formulation")
endif()
configure_file("${checkpoint}"
  "${test_directory}/resumed/restart_1.dat" COPYONLY)
configure_file("${metadata}"
  "${test_directory}/resumed/restart_1.dat.layout.json" COPYONLY)
 
# ep.cpp derives start_step=1 from restart_1.dat, while -time provides .05.
# TS max_steps is the final total step count, so the resumed run still uses 2.
run_auxiliary_restart_stage(resumed ""
  -restart restart_1.dat -time 0.05 -write_restart 0
  -elastic_ts_dt 0.05 -elastic_ts_max_time 0.1 -elastic_ts_max_steps 2
  ${final_vector_option})
execute_process(
  COMMAND "${EP_MPI_RUN}" ${EP_MPI_RUN_FLAGS} -np "${EP_RANKS}"
    "${EP_COMPARATOR}"
    -compare_left "${test_directory}/uninterrupted/final.vec"
    -compare_right "${test_directory}/resumed/final.vec"
    -compare_count 1 -compare_tolerance "${EP_COMPARE_TOLERANCE}"
  RESULT_VARIABLE comparison_result
  OUTPUT_FILE "${test_directory}/comparison.log"
  ERROR_FILE "${test_directory}/comparison.log"
  TIMEOUT 60)
if(NOT comparison_result STREQUAL "0")
  file(READ "${test_directory}/comparison.log" comparison_log)
  message(FATAL_ERROR "Auxiliary restart comparison failed:\n${comparison_log}")
endif()
 
# Invalid vector bytes establish that metadata is checked before VecLoad.
# The original checkpoint and its metadata remain intact in initial/.
foreach(stage missing_metadata wrong_formulation)
  file(WRITE "${test_directory}/${stage}/restart_1.dat"
    "This fixture is deliberately not a PETSc vector.\n")
endforeach()
file(REMOVE "${test_directory}/missing_metadata/restart_1.dat.layout.json")
string(JSON wrong_layout SET "${layout}" formulation "\"direct_stretch\"")
file(WRITE "${test_directory}/wrong_formulation/restart_1.dat.layout.json"
  "${wrong_layout}\n")
run_auxiliary_restart_stage(missing_metadata
  "Auxiliary logarithmic-stress restart requires"
  -restart restart_1.dat -time 0.05 -write_restart 0)
run_auxiliary_restart_stage(wrong_formulation
  "Restart layout mismatch for 'formulation'"
  -restart restart_1.dat -time 0.05 -write_restart 0)
 
message(STATUS
  "Auxiliary ${EP_RANKS}-rank restart matches the uninterrupted solve; missing and incompatible metadata fail before vector loading")