o "i @s6dZddlmZddlmZmZmZddlZddlm Z ddlm Z gd Z ee e edfZd efd d Zd efd dZd efddZd efddZd&ded dfddZGdddZGdddZeZead&ded efddZGdddeZded efddZd efd d!Zded dfd"d#Zd e fd$d%ZdS)'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 StreamContextEventreturncC tjjS)z/Returns a bool indicating if CPU supports AVX2.)torch_C_cpu_is_cpu_support_avx2rrP/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/torch/cpu/__init__.pyr  rcCr)z1Returns a bool indicating if CPU supports AVX512.)rrr_is_cpu_support_avx512rrrrr%rrcCr)z/Returns a bool indicating if CPU supports VNNI.)rrr_is_cpu_support_vnnirrrrr*rrcCdS)zReturns a bool indicating if CPU is currently available. N.B. This function only exists to facilitate device-agnostic code Trrrrrr /sr rcCr )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. Nrrrrrr 8r c@s.eZdZdZd deddfddZd dd ZdS) rzH N.B. This class only exists to facilitate device-agnostic code priorityrNcCdSNr)selfr#rrr__init__GzStream.__init__cCr$r%rr&rrrr wait_streamJr(zStream.wait_stream)r"rN)__name__ __module__ __qualname____doc__intr'r*rrrrrBsrc@s<eZdZdefddZd d ddZd ddZd d d d ZdS) rrcCr )NTrr&rrrqueryOr(z Event.queryNcCr$r%rr)rrrrecordRr(z Event.recordcCr$r%rr1rrrr Ur(zEvent.synchronizecCr$r%rr)rrrwaitXr(z Event.waitr%r+)r,r-r.boolr2r3r r4rrrrrNs   rcCstS)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_streamrrrrr`s rc@sHeZdZUdZeeed<ddZddZde de d e d d fd d Z d S)rzvContext-manager that selects a given stream. N.B. This class only exists to facilitate device-agnostic code cur_streamcCs||_t|_dSr%)r_default_cpu_stream prev_streamr)rrrr'us zStreamContext.__init__cCs |j}|dur dSt|_|adSr%)rr6r9)r&r7rrr __enter__ys zStreamContext.__enter__typevalue tracebackrNcCs|j}|dur dS|jadSr%)rr9r6)r&r;r<r=r7rrr__exit__s zStreamContext.__exit__) r,r-r.r/rr__annotations__r'r:rr>rrrrrls   rrcCst|S)zWrapper around the Context-manager StreamContext that selects a given stream. N.B. This function only exists to facilitate device-agnostic code )r)rrrrrscCr )zReturns number of CPU devices (not cores). Always 1. N.B. This function only exists to facilitate device-agnostic code r rrrrrrrcCr )zzSets the current device, in CPU we do nothing. N.B. This function only exists to facilitate device-agnostic code Nrrrrrrr!rcCr )zyReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrrrrr r@r r%) r/ contextlibrtypingrrrrr_devicer __all__strr0 _device_tr5rrrr r rrr8r6rrrrrr rrrrs.