Memory Machine Cloud agent example usage#

This example shows how to use the MMCloud agent to execute tasks on MemVerge Memory Machine Cloud.

from flytekit import Resources, task, workflow
from flytekitplugins.mmcloud import MMCloudConfig

MMCloudConfig configures MMCloudTask. Tasks specified with MMCloudConfig will be executed using MMCloud. Tasks will be executed with requests cpu="1" and mem="1Gi" by default.

@task(task_config=MMCloudConfig())
def to_str(i: int) -> str:
    return str(i)


@task(task_config=MMCloudConfig())
def to_int(s: str) -> int:
    return int(s)

Resource (cpu and mem) requests and limits, container images, and environment variable specifications are supported.

@task(
    task_config=MMCloudConfig(submit_extra="--migratePolicy [enable=true]"),
    requests=Resources(cpu="1", mem="1Gi"),
    limits=Resources(cpu="2", mem="4Gi"),
    environment={"KEY": "value"},
)
def concatenate_str(s1: str, s2: str) -> str:
    return s1 + s2


@workflow
def concatenate_int_wf(i1: int, i2: int) -> int:
    i1_str = to_str(i=i1)
    i2_str = to_str(i=i2)
    return to_int(s=concatenate_str(s1=i1_str, s2=i2_str))