Skip to content

AcceleratorPartitionConfig

Source code in docs_src/capo_sagemaker/types/accelerator_partition_config.py
class AcceleratorPartitionConfig(TypedDict, closed=True):
    type: NotRequired["capo_sagemaker.types.mig_profile_type.MIGProfileType"]
    """<p>The Multi-Instance GPU (MIG) profile type that defines the partition configuration. The profile specifies the compute and memory allocation for each partition instance. The available profile types depend on the instance type specified in the compute quota configuration.</p>"""
    count: NotRequired["capo_sagemaker.types.integer.Integer"]
    """<p>The number of accelerator partitions to allocate with the specified partition type. If you don't specify a value for vCPU and MemoryInGiB, SageMaker AI automatically allocates ratio-based values for those parameters based on the accelerator partition count you provide.</p>"""

count instance-attribute

count: NotRequired[Integer]

The number of accelerator partitions to allocate with the specified partition type. If you don't specify a value for vCPU and MemoryInGiB, SageMaker AI automatically allocates ratio-based values for those parameters based on the accelerator partition count you provide.

type instance-attribute

type: NotRequired[MIGProfileType]

The Multi-Instance GPU (MIG) profile type that defines the partition configuration. The profile specifies the compute and memory allocation for each partition instance. The available profile types depend on the instance type specified in the compute quota configuration.