fj- " d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZddlm Z m
Z
ddl m Z dgZ
ej dk rd dlZ G d d e Zn% G d
d e Zd Zd Z e j ee G d
de Z G d de ZdS ) N) defaultdict ) reductionassert_spawning)util
BufferWrapperwin32c @ e Zd Z ej Zd Zd Zd ZdS )Arenac t || _ t d D ]p}dt j t | j fz }t
j d|| }t j dk r n$| qt d || _ || _ | j | j f| _
d S )Nd z pym-%d-%stagnamer z Cannot find name for new mmap)sizerangeosgetpidnext_randmmap_winapi GetLastErrorcloseFileExistsErrornamebuffer_state)selfr ir bufs ;/opt/alt/python311/lib64/python3.11/multiprocessing/heap.py__init__zArena.__init__& s DI3ZZ
G
G"bikk4
3C3C%DDiD$777'))Q..E %&EFFF DI DK 9di0DKKK c . t | | j S N)r r )r s r" __getstate__zArena.__getstate__5 s D ! ! !; r$ c ~ |x\ | _ | _ | _ t j d| j | j | _ d S )Nr r )r r r r r )r states r" __setstate__zArena.__setstate__9 s7 16 6 DIty4; )B 49EEEDKKKr$ N) __name__
__module__ __qualname__tempfile_RandomNameSequencer r# r' r* r$ r" r r s] -,..
1
1
1 F F F F Fr$ r c @ e Zd Z ej dk rdgZng ZddZd ZdS )r linuxz/dev/shmr c || _ || _ |dk rt j dt j z | | \ | _ }t j | t j | t j
| j f t j | j | t j | j | j | _
d S )Nr zpym-%d-)prefixdir)r fdr. mkstempr r _choose_dirunlinkr Finalizer ftruncater r )r r r6 r s r" r# zArena.__init__M s DIDGRxx !) 0 %bikk 1 ))$//!1 !1 !1
$
dBHtwj999 TWd+++ )DGTY77DKKKr$ c | j D ]-}t j | }|j |j z |k r|c S .t j S r& )_dir_candidatesr statvfsf_bavailf_frsizer get_temp_dir)r r dsts r" r8 zArena._choose_dir[ sT )
Z]];,44 HHH 5$&& &r$ N)r )r+ r, r- sysplatformr= r# r8 r0 r$ r" r r C sY <7 " " )lOO O 8 8 8 8 ' ' ' ' 'r$ c | j dk rt d t | j t j | j ffS )Nr zDArena is unpicklable because forking was enabled when it was created)r6
ValueError
rebuild_arenar r DupFd)as r" reduce_arenarK d sF 42:: G H H
H qvyqt'<'< ===r$ c F t | | S r& )r detach)r dupfds r" rH rH j s T5<<>>***r$ c z e Zd ZdZdZdZej fdZe d Z
d Z d Z d Z
d Zd Zd
Zd Zd Zd
ZdS )Heap i @ c t j | _ t j | _ || _ g | _ i | _ i | _ i | _
t t | _
g | _ g | _ d| _ d| _ d S Nr )r r _lastpid threadingLock_lock_size_lengths _len_to_seq_start_to_block_stop_to_blockr set_allocated_blocks_arenas_pending_free_blocks
_n_mallocs_n_frees)r r s r" r# z
Heap.__init__{ s
^%%
" ! "-S!1!1 %'!
r$ c |dz
}| |z | z S )Nr r0 )n alignmentmasks r" _roundupz
Heap._roundup s 1}DTE!!r$ c . | t | j | t j }| j | j k r| xj dz c_ t
j d| t | }| j
| |d|fS )N z"allocating a new mmap of length %dr ) rg maxrX r PAGESIZE_DOUBLE_ARENA_SIZE_UNTILr infor r_ append)r r lengtharenas r"
_new_arenazHeap._new_arena s s4:t 4 4dmDD :5 5 5 JJ!OJJ 6???f
E"""q&!!r$ c T |j }|| j k rd S | j | }| j |df= | j ||f= | j | | j | }| |d|f |s$| j |= | j | d S d S rS )
r _DISCARD_FREE_SPACE_LARGER_THANr^ popr[ r\ r_ removerZ rY )r rp ro blocksseqs r" _discard_arenazHeap._discard_arena s D8 8 8 F'++E22 % , 0 E""" v&
E1f%&&& ) ( M ( ( ( ( ( ) )r$ c H t j | j | }|t | j k r| | S | j | }| j | }| }|s| j |= | j |= |\ }}}| j ||f= | j ||f= |S r& ) bisect bisect_leftrY lenrq rZ rt r[ r\ ) r r r ro rw blockrp startstops r" _mallocz Heap._malloc s t}d 3 3 DM"" " "??4(( (]1%F"6*CGGIIE
?$V,dmA.> $ t % 0
. r$ c |\ }}} | j ||f }| | \ }}n# t $ r Y nw xY w | j ||f }| | \ }}n# t $ r Y nw xY w|||f}||z
} | j | | n5# t $ r( |g| j |< t
j | j | Y nw xY w|| j ||f< || j ||f< d S r& ) r\ _absorbKeyErrorr[ rZ rn rz insortrY ) r r} rp r~ r
prev_block_
next_blockro s r" _add_free_blockzHeap._add_free_block sY $ t 0 ,eU^<J ||J//HE11 D
/ -udm<J ll:..GAtt D
t$ 1 V $ + +E 2 2 2 2 1 1 1(-wD V $ M$- 0 0 0 0 0 1 05 eU^,-2UDM***s0 0
==A) )
A65A6 B% %/CCc |\ }}}| j ||f= | j ||f= ||z
}| j | }| | |s"| j |= | j | ||fS r& )r[ r\ rZ ru rY )r r} rp r~ r ro rw s r" r z Heap._absorb s $ t % 0
. v&
5 ) ( M ( ( (d{r$ c |\ }}}| j | }| ||f |s| | d S d S r& )r^ ru rx )r r} rp r~ r rv s r" _remove_allocated_blockz Heap._remove_allocated_block s] "ud'.
udm$$$ ' & & & & & ' 'r$ c | j }n# t $ r Y d S w xY w| | | | Wr& )r` rt
IndexErrorr r r r} s r" _free_pending_blockszHeap._free_pending_blocks sq 0
15577
' ' ' ( ( / / /
0s
++c ( t j | j k r9t d t j | j | j d s | j | d S | xj dz
c_ |
| | | | | j
d S # | j
w xY w)Nz$My pid ({0:n}) is not last pid {1:n}Fr )r r rT rG formatrW acquirer` rn rb r r r releaser s r" freez Heap.free s 9;;$- ' ' 6==IKK
/ /0 0
0 z!!%(( %
% , ,U 3 3 3 3 3
%
"
))+++$$U+++,,U333
""$$$$$
""$$$$s
AC6 6Dc |dk r"t d | t j |k r"t d | t j | j k r| | j 5 | xj
dz
c_
| | t |d | j }| | \ }}}||z }||k r| |||f | j | ||f |||fcd d d S # 1 swxY w Y d S )Nr Size {0:n} out of rangeSize {0:n} too larger )rG r rD maxsize
OverflowErrorr r rT r# rW ra r rg rj
_alignmentr r r^ add)r r rp r~ r real_stops r" mallocz Heap.malloc( s !88 6 = =d C CDD D ;$ 6 = =d C CDD D
9;;$- ' ' MMOOO
Z - - OOq OO % % ' ' '==T1 t??D#'<<#5#5 UE4 I4 $$eY%=>>> "5 ) - -ui.@ A A A5), - - - - - - - - - - - - - - - - - -s B0EEEN)r+ r, r- r rs rl r rk r# staticmethodrg rq rx r r r r r r r r0 r$ r" rP rP s s J&3# , M : " " \"
"
"
") ) )$ " 3 3 3<
' ' '0 0 0% % %8- - - - -r$ rP c . e Zd Z e Zd Zd ZdS )r c X |dk r"t d | t j |k r"t d | t
j | }||f| _ t j
| t
j j |f d S )Nr r r )args) rG r rD r r r _heapr r r r: r )r r r} s r" r# zBufferWrapper.__init__F s !88 6 = =d C CDD D ;$ 6 = =d C CDD D #**400 dm
dM / 4E8DDDDDDr$ c \ | j \ \ }}}}t |j |||z S r& )r
memoryviewr )r rp r~ r r s r" create_memoryviewz BufferWrapper.create_memoryviewO s3 %)[" td%,''eDj(899r$ N)r+ r, r- rP r r# r r0 r$ r" r r B s@ DFFEE E E: : : : :r$ )rz collectionsr r r rD r. rU contextr r r __all__rE r objectr rK rH registerrP r r0 r$ r" <module>r s
# # # # # #
/ / / / / / / /
<7NNN F F F F F F F F FH ' ' ' ' ' ' ' 'B> > > + + + Iul+++ I- I- I- I- I-6 I- I- I-^: : : : :F : : : : :r$