&VjidZddlmZddlmZmZmZddlZddlm Z ddlm Z gd Z d e fd Zd e fd Zd e fd Zd e fdZd e fdZd e fdZd e fdZd e fdZd"dejjd dfdZGddZGddZeZead"dejjd efdZGddeZded efdZd e fdZ!dejjd dfdZ"d e#fd Z$d e fd!Z%dS)#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 StreamContextEventreturncHtjjS)z/Returns a bool indicating if CPU supports AVX2.)torch_C_cpu_is_avx2_supportedL/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/cpu/__init__.pyrrs 8= + + - --rcHtjjS)z1Returns a bool indicating if CPU supports AVX512.)rrr_is_avx512_supportedrrrr!r!$s 8= - - / //rcHtjjS)z6Returns a bool indicating if CPU supports AVX512_BF16.)rrr_is_avx512_bf16_supportedrrrr#r#)s 8= 2 2 4 44rcHtjjS)z/Returns a bool indicating if CPU supports VNNI.)rrr_is_avx512_vnni_supportedrrr_is_vnni_supportedr&.s 8= 2 2 4 44rcHtjjS)z3Returns a bool indicating if CPU supports AMX_TILE.)rrr_is_amx_tile_supportedrrrr(r(4 8= / / 1 11rcHtjjS)z3Returns a bool indicating if CPU supports AMX FP16.)rrr_is_amx_fp16_supportedrrrr+r+9r)rcHtjjS)zInitializes AMX instructions.)rrr _init_amxrrrr-r->s 8= " " $ $$rcdS)zReturns a bool indicating if CPU is currently available. N.B. This function only exists to facilitate device-agnostic code Trrrrr r Cs 4rr cdS)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 rrrLrc<eZdZdZd deddfdZd dZd dZd d ZdS) rzH N.B. This class only exists to facilitate device-agnostic code priorityrNcdSNr)selfr3s r__init__zStream.__init__[ rcdSr5rr6rs r wait_streamzStream.wait_stream^r8rcdSr5rr6s r record_eventzStream.record_eventar8rcdSr5r)r6events r wait_eventzStream.wait_eventdr8r)r2rN) __name__ __module__ __qualname____doc__intr7r;r>rArrrrrVs   d                  rrc6eZdZdefdZdddZddZdddZdS) rrcdS)NTrr=s rqueryz Event.queryistrNcdSr5rr:s rrecordz Event.recordlr8rcdSr5rr=s rrzEvent.synchronizeor8rcdSr5rr:s rwaitz Event.waitrr8rr5rB)rCrDrEboolrJrLrrOrrrrrhsut                rrctS)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  rcNeZdZUdZeeed<dZdZde de de dd fd Z d S) rzvContext-manager that selects a given stream. N.B. This class only exists to facilitate device-agnostic code cur_streamc,||_t|_dSr5)r_default_cpu_stream prev_streamr:s rr7zStreamContext.__init__s .rc8|j}|dSt|_|adSr5)rrRrW)r6rTs r __enter__zStreamContext.__enter__s'[   F+$rtypevalue tracebackrNc*|j}|dS|jadSr5)rrWrR)r6rZr[r\rTs r__exit__zStreamContext.__exit__s"[   F*r) rCrDrErFrr__annotations__r7rYrr^rrrrrs     ///%%%+S+++++++++rrrc t|S)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   rcdS)zReturns number of CPU devices (not cores). Always 1. N.B. This function only exists to facilitate device-agnostic code r rrrrrrs 1rcdS)zzSets the current device, in CPU we do nothing. N.B. This function only exists to facilitate device-agnostic code Nrrs rrrr0rcdS)zyReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrrrrrs 5rcdS)zReturns True if the CPU is initialized. Always True. N.B. This function only exists to facilitate device-agnostic code Trrrrr r s 4rr5)&rF contextlibrtypingrrrrr _devicer __all__rPrr!r#r&r(r+r-r typesDevicerrrrVrRrrrrGrrstrrr rrrrns .-----''''''''''    .D.... 0d0000 545555 5D5555 22222 22222 %4%%%% d *d        $         fhh%  5;-     +++++*+++>!6!4!!!! c    u{)dr