9iSrSSKJr SSKJrJrJr SSKrSSKJ r SSKJ r /S Qr S \ 4S jrS \ 4S jrS \ 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 \ 4Sjr!S\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)AnyOptionalUnionNdevice)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_supportedR/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/torch/cpu/__init__.pyrrs 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(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 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 rrrLrcL\rSrSrSrS S\SS4SjjrS SjrS SjrS S jr S r g) rVz@ N.B. This class only exists to facilitate device-agnostic code priorityrNcgNr)selfr3s r__init__Stream.__init__[ rcgr5rr6rs r wait_streamStream.wait_stream^r9rcgr5rr6s r record_eventStream.record_eventar9rcgr5r)r6events r wait_eventStream.wait_eventdr9rr)rN) __name__ __module__ __qualname____firstlineno____doc__intr7r<r@rD__static_attributes__rrrrrVs)  d    rrcH\rSrSrS\4SjrS S SjjrS SjrS S SjjrSr g) rhrcg)NTrr?s rquery Event.queryisrNcgr5rr;s rrecord Event.recordlr9rcgr5rr?s rrEvent.synchronizeor9rcgr5rr;s rwait Event.waitrr9rrr5rG) rHrIrJrKboolrRrUrrZrNrrrrrhs"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[Ulgr5)r_default_cpu_stream prev_streamr;s rr7StreamContext.__init__s .rc>URnUcg[UlUqgr5)rr^rd)r6ras r __enter__StreamContext.__enter__s#[[   +$rtypevalue tracebackrNc<URnUcgURqgr5)rrdr^)r6rirjrkras r__exit__StreamContext.__exit__s![[   **r)rdr) rHrIrJrKrLrr__annotations__r7rgrrmrNrrrrrs?   /%+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 rrrr0rcg)zqReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrrrrrs rcg)zxReturns True if the CPU is initialized. Always True. N.B. This function only exists to facilitate device-agnostic code Trrrrr r s rr5)&rL contextlibrtypingrrrrr _devicer __all__r\rr!r#r&r(r+r-r typesDevicerrrrcr^rrrrMrrstrrr rrrr~sD .''  .D. 0d0 545 5D5 22 22 %4% d **d  $   h% 5;;--  +*+>!6!4! c u{{))dr