IЦiJSrSSKJr SSKJrJrJr SSKrSSKJ r SSKJ r /S Qr \\ \ \S4rS \4S jrS \4S jrS \4S jrS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS"S\S S4Sjjr"SS5r"SS5r\"5r\qS"S\S \4Sjjr"SS\5r S\S \4Sjr!S \4Sjr"S\S S4S jr#S \ 4S!jr$g)#zj This package implements abstractions found in ``torch.cuda`` to facilitate writing device-agnostic code. )AbstractContextManager)AnyOptionalUnionNdevice)amp) is_available synchronizecurrent_devicecurrent_streamstream set_device device_countStream StreamContextEventreturncR[RRR5$)z/Returns a bool indicating if CPU supports AVX2.)torch_C_cpu_is_avx2_supportedQ/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/cpu/__init__.pyrr s 88== + + --rcR[RRR5$)z1Returns a bool indicating if CPU supports AVX512.)rrr_is_avx512_supportedrrrr r %s 88== - - //rcR[RRR5$)z6Returns a bool indicating if CPU supports AVX512_BF16.)rrr_is_avx512_bf16_supportedrrrr"r"*s 88== 2 2 44rcR[RRR5$)z/Returns a bool indicating if CPU supports VNNI.)rrr_is_avx512_vnni_supportedrrr_is_vnni_supportedr%/s 88== 2 2 44rcR[RRR5$)z3Returns a bool indicating if CPU supports AMX_TILE.)rrr_is_amx_tile_supportedrrrr'r'5 88== / / 11rcR[RRR5$)z3Returns a bool indicating if CPU supports AMX FP16.)rrr_is_amx_fp16_supportedrrrr*r*:r(rcR[RRR5$)zInitializes AMX instructions.)rrr _init_amxrrrr,r,?s 88== " " $$rcR[RRR5$)z2Returns a bool indicating if CPU supports Arm SVE.)rrr_is_arm_sve_supportedrrrr.r.Ds 88== . . 00rcg)z}Returns a bool indicating if CPU is currently available. N.B. This function only exists to facilitate device-agnostic code Trrrrr r Is rr cg)zWaits for all kernels in all streams on the CPU device to complete. Args: device (torch.device or int, optional): ignored, there's only one CPU device. N.B. This function only exists to facilitate device-agnostic code. Nrrs rr r Rrc8\rSrSrSrS S\SS4SjjrS SjrSrg) r\z@ N.B. This class only exists to facilitate device-agnostic code priorityrNcgNr)selfr4s r__init__Stream.__init__a rcgr6rr7rs r wait_streamStream.wait_streamdr:rr)rN) __name__ __module__ __qualname____firstlineno____doc__intr8r=__static_attributes__rrrrr\s  d  rrcH\rSrSrS\4SjrS S SjjrS SjrS S SjjrSr g) rhrcg)NTrr7s rquery Event.queryisrNcgr6rr<s rrecord Event.recordlr:rcgr6rrKs rr Event.synchronizeor:rcgr6rr<s rwait Event.waitrr:rrr6r@) rArBrCrDboolrLrOr rTrGrrrrrhs"t    rrc[$)zReturns the currently selected :class:`Stream` for a given device. Args: device (torch.device or int, optional): Ignored. N.B. This function only exists to facilitate device-agnostic code )_current_streamrs rrrzs  rcP\rSrSr%Sr\\\S'SrSr S\ S\ S\ S S 4S jr S r g ) rznContext-manager that selects a given stream. N.B. This class only exists to facilitate device-agnostic code cur_streamc&Xl[Ulgr6)r_default_cpu_stream prev_streamr<s rr8StreamContext.__init__s .rc>URnUcg[UlUqgr6)rrXr^)r7r[s r __enter__StreamContext.__enter__s#[[   +$rtypevalue tracebackrNc<URnUcgURqgr6)rr^rX)r7rcrdrer[s r__exit__StreamContext.__exit__s![[   **r)r^r) rArBrCrDrErr__annotations__r8rarrgrGrrrrrs?   /%+S++++rrrc[U5$)zWrapper around the Context-manager StreamContext that selects a given stream. N.B. This function only exists to facilitate device-agnostic code )r)rs rrrs   rcg)zxReturns number of CPU devices (not cores). Always 1. N.B. This function only exists to facilitate device-agnostic code r rrrrrrs rcg)zrSets the current device, in CPU we do nothing. N.B. This function only exists to facilitate device-agnostic code Nrrs rrrr1rcg)zqReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrrrrrs rr6)%rE contextlibrtypingrrrrr _devicer __all__strrF _device_trVrr r"r%r'r*r,r.r r rrr]rXrrrrrrrrrrvs; .''   '3T) * .D. 0d0 545 5D5 22 22 %4% 1t1 d T      h% 9  +*+>!6!4! c yTr