o bi/,@sddlZddlZddlmZddlmZddlZddlmm Z ddl m Z e e ZdZedZGdddedgd ZdS) N) namedtuple)Optional)RESOURCE_CONSTRAINT_PREFIXznode:__internal_head__csTeZdZdZ     d fdd ZddZddZdd ed ee fd d Z Z S) ResourceSpecaRepresents the resource configuration passed to a raylet. All fields can be None. Before starting services, resolve() should be called to return a ResourceSpec with unknown values filled in with defaults based on the local machine specifications. Attributes: num_cpus: The CPUs allocated for this raylet. num_gpus: The GPUs allocated for this raylet. memory: The memory allocated for this raylet. object_store_memory: The object store memory allocated for this raylet. Note that when calling to_resource_dict(), this will be scaled down by 30% to account for the global plasma LRU reserve. resources: The custom resources allocated for this raylet. Ncstt|||||||SN)superr__new__)clsnum_cpusnum_gpusmemoryobject_store_memory resources __class__N/home/ubuntu/.local/lib/python3.10/site-packages/ray/_private/resource_spec.pyr /s zResourceSpec.__new__cCs$|D] }|durdSqdS)z;Returns if this ResourceSpec has default values filled out.NFT)_asdictvalues)selfvrrrresolved@s zResourceSpec.resolvedcCs|sJt|j|j|jt|jt|jd}dd|D}|D]H\}}t |tsAt |t sAJ|dt |d|t |t rR| sRt d|||dkr^t d|||tjkrmt d tj||q%|S) zReturns a dict suitable to pass to raylet initialization. This renames num_cpus / num_gpus to "CPU" / "GPU", translates memory from bytes into 100MB memory units, and checks types. )CPUGPUr rcSsi|] \}}|dkr||qS)rr).0resource_labelresource_quantityrrr Ws z1ResourceSpec.to_resource_dict..z (z): zOResource quantities must all be whole numbers. Violated by resource '{}' in {}.rzIResource quantities must be nonnegative. Violated by resource '{}' in {}.zHResource quantities must be at most {}. Violated by resource '{}' in {}.)rdictrr r intr ritems isinstancefloattype is_integer ValueErrorformat ray_constantsMAX_RESOURCE_QUANTITY)rrrrrrrto_resource_dictGsN   zResourceSpec.to_resource_dictis_headnode_ip_addressc Cs|jpi}d|vsJ|d|vsJ|d|vsJ|d|vs'J||dur0tj}d|t|<t|vrAttd|rGd|t<|j}|durTtj j }d}tj j D]}tj j |}d}|dkro|j}n||d}|} |dur| dur|t| krtd |d |d |d | d |dur|}| durt|t| }|r|dkr|}n|||<|} | rd|t| <ddlm} | | |} | r|| q\tjj } tj j }|j }|dur>t!|t"j#}t$j%dkrt|t"j&}t"j'}t$j%dkst$j%dkr(tj j (d}t)t"j*|}t||}|r>||kr>t+,d-|dd|}|j.}|duri||}|dkri|d| kritd-t/|ddt!d|| t0|||||}|1sxJ|S)aReturns a copy with values filled out with system defaults. Args: is_head: Whether this is the head node. node_ip_address: The IP address of the node that we are on. This is used to automatically create a node id resource. rrr rNg?z7 is a reserved resource name, use another name instead.rz Attempting to start raylet with  z, but z contains .) usage_libdarwinlinuxlinux2gffffff?z.Warning: Capping object memory store to {}GB. geAz]To increase this further, specify `object_store_memory` when calling ray.init() or ray start.gחAg?zAfter taking into account object store and redis memory usage, the amount of memory on this node available for tasks and actors ({} GB) is less than {}% of total. You can adjust these settings with ray.init(memory=, object_store_memory=).d)2rcopyrayutilget_node_ip_addressNODE_ID_PREFIXHEAD_NODE_RESOURCE_NAMEr&r _privateutils get_num_cpus accelerators"get_all_accelerator_resource_names$get_accelerator_manager_for_resourcer get+get_current_process_visible_accelerator_idslen#get_visible_accelerator_ids_env_var!get_current_node_num_acceleratorsmin!get_current_node_accelerator_typerray._private.usager0record_hardware_usage%get_current_node_additional_resourcesupdate_commonget_system_memoryestimate_available_memoryrr r(&DEFAULT_OBJECT_STORE_MEMORY_PROPORTIONsysplatform!MAC_DEGRADED_PERF_MMAP_SIZE_LIMIT%DEFAULT_OBJECT_STORE_MAX_MEMORY_BYTESget_shared_memory_bytesmaxREQUIRE_SHM_SIZE_THRESHOLDloggerdebugr'r roundrr)rr+r,rr r accelerator_resource_nameaccelerator_managernum_acceleratorsvisible_accelerator_idsaccelerator_typer0additional_resources system_memory avail_memoryrobject_store_memory_cap shm_availshm_capr specrrrresolve{s                   zResourceSpec.resolve)NNNNNr) __name__ __module__ __qualname____doc__r rr*boolrstrrg __classcell__rrrrrs  4r)r r r rr)loggingrQ collectionsrtypingrr7ray._private.ray_constantsr<r(ray._common.utilsr getLoggerrhrXr:r;rrrrrs