Skip to content

Commit

Permalink
Upgrade Slurm version to 23.02.05
Browse files Browse the repository at this point in the history
Also update test for pmix upgrade to 4.2.6

Signed-off-by: Nicola Sirena <[email protected]>
  • Loading branch information
NSsirena committed Sep 12, 2023
1 parent 84fa175 commit 13e0e67
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/integration-tests/tests/schedulers/test_slurm.py
Original file line number Diff line number Diff line change
Expand Up @@ -229,18 +229,18 @@ def test_slurm_pmix(pcluster_config_reader, scheduler, clusters_factory, use_log
remote_command_executor = RemoteCommandExecutor(cluster, use_login_node=use_login_node)

# Ensure the expected PMIx version is listed when running `srun --mpi=list`.
# Since we're installing PMIx v3.1.5, we expect to see pmix and pmix_v3 in the output.
# Since we're installing PMIx v4.2.6, we expect to see pmix and pmix_v4 in the output.
# Sample output:
# [ec2-user@ip-172-31-33-187 ~]$ srun 2>&1 --mpi=list
# srun: MPI types are...
# srun: none
# srun: openmpi
# srun: pmi2
# srun: pmix
# srun: pmix_v3
# srun: pmix_v4
mpi_list_output = remote_command_executor.run_remote_command("srun 2>&1 --mpi=list").stdout
assert_that(mpi_list_output).matches(r"\s+pmix($|\s+)")
assert_that(mpi_list_output).matches(r"\s+pmix_v3($|\s+)")
assert_that(mpi_list_output).matches(r"\s+pmix_v4($|\s+)")

# Compile and run an MPI program interactively
mpi_module = "openmpi"
Expand Down Expand Up @@ -1711,7 +1711,7 @@ def _gpu_resource_check(slurm_commands, partition, instance_type, instance_type_
def _test_slurm_version(remote_command_executor):
logging.info("Testing Slurm Version")
version = remote_command_executor.run_remote_command("sinfo -V").stdout
assert_that(version).is_equal_to("slurm 23.02.4")
assert_that(version).is_equal_to("slurm 23.02.5")


def _test_job_dependencies(slurm_commands, region, stack_name, scaledown_idletime):
Expand Down

0 comments on commit 13e0e67

Please sign in to comment.