ȅiSrSSKJr SSKJr SSKrSSKJr SSKJ r /S Qr S \ 4S jr S \ 4S jr S \ 4S jrS \ 4SjrS \ 4SjrS \ 4SjrS \ 4SjrS \ 4SjrS"S\R(R*S S4Sjjr"SS5r"SS5r\"5r\qS"S\R(R*S \4Sjjr"SS\5rS\S \4SjrS \4SjrS\R(R*S S4Sjr S \!4S jr"S \ 4S!jr#g)#zj This package implements abstractions found in ``torch.cuda`` to facilitate writing device-agnostic code. )AbstractContextManager)AnyNdevice)amp) is_availableis_initialized synchronizecurrent_devicecurrent_streamstream set_device device_countStream StreamContextEventreturncR[RRR5$)z/Returns a bool indicating if CPU supports AVX2.)torch_C_cpu_is_avx2_supportedL/home/james-whalen/.local/lib/python3.13/site-packages/torch/cpu/__init__.pyrrs 88== + + --rcR[RRR5$)z1Returns a bool indicating if CPU supports AVX512.)rrr_is_avx512_supportedrrrrr$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&4 88== / / 11rcR[RRR5$)z3Returns a bool indicating if CPU supports AMX FP16.)rrr_is_amx_fp16_supportedrrrr)r)9r'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 Cs rrcg)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 LrcL\rSrSrSrS S\SS4SjjrS SjrS SjrS S jr S r g) rVz@ N.B. This class only exists to facilitate device-agnostic code priorityrNcgNr)selfr1s r__init__Stream.__init__[ rcgr3rr4rs r wait_streamStream.wait_stream^r7rcgr3rr4s r record_eventStream.record_eventar7rcgr3r)r4events r wait_eventStream.wait_eventdr7rr)rN) __name__ __module__ __qualname____firstlineno____doc__intr5r:r>rB__static_attributes__rrrrrVs)  d    rrcH\rSrSrS\4SjrS S SjjrS SjrS S SjjrSr g) rhrcg)NTrr=s rquery Event.queryisrNcgr3rr9s rrecord Event.recordlr7rcgr3rr=s rr Event.synchronizeor7rcgr3rr9s rwait Event.waitrr7rrr3rE) rFrGrHrIboolrPrSr rXrLrrrrrhs"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-\S'SrSrS\ S\ S \ S S4S jr S r g) rznContext-manager that selects a given stream. N.B. This class only exists to facilitate device-agnostic code N cur_streamc&Xl[Ulgr3)r_default_cpu_stream prev_streamr9s rr5StreamContext.__init__s .rc>URnUcg[UlUqgr3)rr\rb)r4r_s r __enter__StreamContext.__enter__s#[[   +$rtypevalue tracebackrc<URnUcgURqgr3)rrbr\)r4rgrhrir_s r__exit__StreamContext.__exit__s![[   **r)rbr) rFrGrHrIrJr__annotations__r5rerrkrLrrrrrs>  /%+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 rrrrrrrs 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 cpurrrrr r s rcg)zxReturns True if the CPU is initialized. Always True. N.B. This function only exists to facilitate device-agnostic code Trrrrr r s rr3)$rJ contextlibrtypingrrr_devicer __all__rZrrr!r$r&r)r+r typesDevicer rrrar\rrrrKrrstrr r rrrr|sA .  .D. 0d0 545 5D5 22 22 %4% d **d  $   h% 5;;--  +*+>!6!4! c u{{))dr