OLCF Frontier¶
Link to official documentation
- class flow.environments.incite.FrontierEnvironment[source]¶
Bases:
DefaultSlurmEnvironment
Environment profile for the Frontier supercomputer.
{# Templated in accordance with: https://docs.olcf.ornl.gov/systems/crusher_quick_start_guide.html #}
{% extends "slurm.sh" %}
{% block tasks %}
{% set threshold = 0 if force else 0.9 %}
{% set cpu_tasks = operations|calc_tasks('np', parallel, force) %}
{% set gpu_tasks = operations|calc_tasks('ngpu', parallel, force) %}
{% set nn = gpu_tasks|calc_num_nodes(cpu_tasks, threshold) %}
{% if gpu_tasks < 1 and not force %}
{% raise "Must request GPUs to use Frontier." %}
{% endif %}
#SBATCH --nodes={{ nn }}
{% endblock tasks %}
{% block header %}
{{- super() -}}
{% set account = account|default(project|get_account_name, true) %}
{% if account %}
#SBATCH --account={{ account }}
{% endif %}
{% endblock header %}