[ThSrSSKJr SSKJrJrJr SSKrSSKJ r SSKJ r /S Qr \\ \ \S4rS \4S jrS \4S jrS \4S jrS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS!S\S S4Sjjr"SS5r"SS5r\"5r\qS!S\S \4Sjjr"SS\5rS\S \4Sjr S \4Sjr!S\S S4Sjr"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_supportedJ/var/www/auris/envauris/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== " " $$rcg)z}Returns a bool indicating if CPU is currently available. N.B. This function only exists to facilitate device-agnostic code Trrrrr r Ds 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 MrcL\rSrSrSrS S\SS4SjjrS SjrS SjrS S jr S r g) rWz@ N.B. This class only exists to facilitate device-agnostic code priorityrNcgNr)selfr2s r__init__Stream.__init__\ rcgr4rr5rs r wait_streamStream.wait_stream_r8rcgr4rr5s r record_eventStream.record_eventbr8rcgr4r)r5events r wait_eventStream.wait_eventer8rr)rN) __name__ __module__ __qualname____firstlineno____doc__intr6r;r?rC__static_attributes__rrrrrWs)  d    rrcH\rSrSrS\4SjrS S SjjrS SjrS S SjjrSr g) rircg)NTrr>s rquery Event.queryjsrNcgr4rr:s rrecord Event.recordmr8rcgr4rr>s rr Event.synchronizepr8rcgr4rr:s rwait Event.waitsr8rrr4rF) rGrHrIrJboolrQrTr rYrMrrrrris"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 rrr{s  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[Ulgr4)r_default_cpu_stream prev_streamr:s rr6StreamContext.__init__s .rc>URnUcg[UlUqgr4)rr]rc)r5r`s r __enter__StreamContext.__enter__s#[[   +$rtypevalue tracebackrNc<URnUcgURqgr4)rrcr])r5rhrirjr`s r__exit__StreamContext.__exit__s![[   **r)rcr) rGrHrIrJrKrr__annotations__r6rfrrlrMrrrrrs?   /%+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 rrrr/rcg)zqReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrrrrrs rr4)$rK contextlibrtypingrrrrr _devicer __all__strrL _device_tr[rr r"r%r'r*r,r r rrrbr]rrrrrrrrrr{s/ .''   '3T) * .D. 0d0 545 5D5 22 22 %4% d T  $   h% 9  +*+>!6!4! c yTr