Skip to content

Commit

Permalink
[integ-tests] Use c5.xlarge in test_slurm
Browse files Browse the repository at this point in the history
Signed-off-by: Francesco De Martino <[email protected]>
  • Loading branch information
demartinofra committed Oct 27, 2021
1 parent a8b4767 commit 498bfb3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/integration-tests/tests/schedulers/test_slurm.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def test_slurm(region, pcluster_config_reader, clusters_factory, test_datadir, a
_test_slurm_version(remote_command_executor)

if supports_impi:
_test_mpi_job_termination(remote_command_executor, test_datadir)
_test_mpi_job_termination(remote_command_executor, test_datadir, instance_type="c5.xlarge")

_assert_no_node_in_cluster(region, cluster.cfn_name, slurm_commands)
_test_job_dependencies(slurm_commands, region, cluster.cfn_name, scaledown_idletime)
Expand Down Expand Up @@ -655,7 +655,7 @@ def _terminate_nodes_manually(instance_ids, region):
logging.info("Terminated nodes: {}".format(instance_ids))


def _test_mpi_job_termination(remote_command_executor, test_datadir):
def _test_mpi_job_termination(remote_command_executor, test_datadir, instance_type):
"""
Test canceling mpirun job will not leave stray processes.
Expand All @@ -669,7 +669,7 @@ def _test_mpi_job_termination(remote_command_executor, test_datadir):

# Submit mpi_job, which runs Intel MPI benchmarks with intelmpi
# Leaving 1 vcpu on each node idle so that the process check job can run while mpi_job is running
result = slurm_commands.submit_script(str(test_datadir / "mpi_job.sh"))
result = slurm_commands.submit_script(str(test_datadir / "mpi_job.sh"), constraint=instance_type)
job_id = slurm_commands.assert_job_submitted(result.stdout)

# Wait for compute node to start and check that mpi processes are started
Expand Down

0 comments on commit 498bfb3

Please sign in to comment.