Skip to content

Commit

Permalink
Fix: CPU features required for execution could not be specified.
Browse files Browse the repository at this point in the history
This allows the user to specify CPU features the CPU that must be available in order to run the VM.

This should help with automatic and manual scheduling to the right machines.
  • Loading branch information
hoh committed May 29, 2024
1 parent 3c0144a commit cec329f
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions aleph_message/models/execution/environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,10 @@ class CpuProperties(HashableModel):
vendor: Optional[Union[Literal["AuthenticAMD", "GenuineIntel"], str]] = Field(
default=None, description="CPU vendor. Allows other vendors."
)
# Features described here differ from the CPU flags (/proc/cpuinfo) since they must be actually available to the VM.
features: Optional[List[str]] = Field(
default=None, description="CPU features required. Examples: 'avx2', 'sev_es', 'sev_snp'.",
)

class Config:
extra = Extra.forbid
Expand Down

0 comments on commit cec329f

Please sign in to comment.