stencil_config
RunMode
Bases: Enum
Run-Mode for the model Build: compile & save compiled files only BuildAndRun: compile & save compiled files, then run Run: load from .so and run, will fail if .so is not available
Build = 0
class-attribute
instance-attribute
BuildAndRun = 1
class-attribute
instance-attribute
Run = 2
class-attribute
instance-attribute
CompilationConfig
backend = backend
instance-attribute
rebuild = rebuild
instance-attribute
validate_args = validate_args
instance-attribute
format_source = format_source
instance-attribute
device_sync = device_sync
instance-attribute
run_mode = run_mode
instance-attribute
use_minimal_caching = use_minimal_caching
instance-attribute
__init__(backend=backend_python, rebuild=True, validate_args=True, format_source=False, device_sync=False, run_mode=RunMode.BuildAndRun, use_minimal_caching=False, communicator=None)
check_communicator(communicator)
Checks that the communicator has a square layout
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
communicator
|
Communicator
|
communicator to use |
required |
Raises:
| Type | Description |
|---|---|
RuntimeError
|
If non-square layout is given |
determine_compiling_equivalent(rank, partitioner)
From my rank & the current partitioner we determine which rank we should read from
get_decomposition_info_from_comm(communicator)
as_dict()
from_dict(data)
classmethod
StencilConfig
dataclass
Bases: Hashable