8hSrSSKJr 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#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_supportedL/var/www/fran/franai/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&0s 88== 2 2 44rcR[RRR5$)z3Returns a bool indicating if CPU supports AMX_TILE.)rrr_is_amx_tile_supportedrrrr(r(6 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 Es 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 rrrNrcL\rSrSrSrS S\SS4SjjrS SjrS SjrS S jr S r g) rXz@ 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_eventcr9rcgr5r)r6events r wait_eventStream.wait_eventfr9rr)rN) __name__ __module__ __qualname____firstlineno____doc__intr7r<r@rD__static_attributes__rrrrrXs)  d    rrcH\rSrSrS\4SjrS S SjjrS SjrS S SjjrSr g) rjrcg)NTrr?s rquery Event.queryksrNcgr5rr;s rrecord Event.recordnr9rcgr5rr?s rrEvent.synchronizeqr9rcgr5rr;s rwait Event.waittr9rrr5rG) rHrIrJrKboolrRrUrrZrNrrrrrjs"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[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__strrM _device_tr\rr!r#r&r(r+r-r rrrrcr^rrrrrrr rrrr}s; .''   '3T) * .D. 0d0 545 5D5 22 22 %4% d T  $   h% 9  +*+>!6!4! c yTr