globus / globus-compute

Globus Compute: High Performance Function Serving for Science
https://www.globus.org/compute
Apache License 2.0
148 stars 47 forks source link

Run tests on Frontera TACC #343

Closed yadudoc closed 3 years ago

yadudoc commented 3 years ago

Launch funcx-endpoint following instructions here: https://github.com/funcx-faas/funcX/tree/forwarder_rearch_1/funcx_sdk/funcx/tests

Part of the test campaign: #318

yadudoc commented 3 years ago

Completed. Here's the config I used:

from funcx_endpoint.endpoint.utils.config import Config
from funcx_endpoint.executors import HighThroughputExecutor
from parsl.providers import SlurmProvider
from parsl.launchers import SrunLauncher
from parsl.addresses import address_by_hostname

user_opts = {
    'frontera': {
        'worker_init': 'source ~/setup_funcx_test_env.sh',
        'scheduler_options': '#SBATCH -A MCB20024',  
        'partition': 'development',

    }
}

config = Config(
    executors=[
        HighThroughputExecutor(
            label="frontera_htex",
            max_workers_per_node=2,
            address=address_by_hostname(),
            provider=SlurmProvider(
                cmd_timeout=60,     # Add extra time for slow scheduler responses
                nodes_per_block=2,
                init_blocks=1,
                min_blocks=0,
                max_blocks=1,
                partition=user_opts['frontera']['partition'],  # Replace with partition name

                # Enter scheduler_options if needed
                scheduler_options=user_opts['frontera']['scheduler_options'], 

                # Command to be run before starting a worker, such as:
                # 'module load Anaconda; source activate parsl_env'.
                worker_init=user_opts['frontera']['worker_init'],

                # Ideally we set the walltime to the longest supported walltime.
                walltime='00:10:00',
                launcher=SrunLauncher(),
            ),
        )
    ],
    funcx_service_address="http://k8s-dev.funcx.org/api/v1",
)