Phb&dZddlmZddlmZmZmZddlZddlm Z ddlm Z gd Z ee e edfZd efd Zd efd Zdd ed dfdZGddZGddZeZeadd ed efdZGddeZded efdZd efdZd ed dfdZd e fdZy)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 StreamContextEventreturncRtjjjS)z/Returns a bool indicating if CPU supports VNNI.)torch_C_cpu_is_cpu_support_vnni]C:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/cpu/__init__.pyrrs 88== - - //rcy)zReturns a bool indicating if CPU is currently available. N.B. This function only exists to facilitate device-agnostic code Trrrrr r #s rr cy)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 ,s rc&eZdZdZddefdZddZy)rzH N.B. This class only exists to facilitate device-agnostic code prioritycyNr)selfr"s r__init__zStream.__init__< rNcyr$rr%rs r wait_streamzStream.wait_stream?r'r))rN)__name__ __module__ __qualname____doc__intr&r*rrrrr7s   rrc.eZdZdefdZddZdZddZy)rrcy)NTrr%s rqueryz Event.queryDsrNcyr$rr)s rrecordz Event.recordGr'rcyr$rr3s rr zEvent.synchronizeJr'rcyr$rr)s rwaitz Event.waitMr'rr$)r,r-r.boolr4r6r r9rrrrrCst   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 rrrUs  rcBeZdZUdZeeed<dZdZde de de fdZ y ) rzvContext-manager that selects a given stream. N.B. This class only exists to facilitate device-agnostic code cur_streamc(||_t|_yr$)r_default_cpu_stream prev_streamr)s rr&zStreamContext.__init__is .rc<|j}|yt|_|ayr$)rr<rA)r%r>s r __enter__zStreamContext.__enter__ms#[[   +$rtypevalue tracebackc:|j}|y|jayr$)rrAr<)r%rDrErFr>s r__exit__zStreamContext.__exit__vs![[   **rN) r,r-r.r/rr__annotations__r&rCrrHrrrrras8   /%+S+++rrrct|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   rcy)zReturns number of CPU devices (not cores). Always 1. N.B. This function only exists to facilitate device-agnostic code r rrrrrrs rcy)zzSets the current device, in CPU we do nothing. N.B. This function only exists to facilitate device-agnostic code Nrrs rrrs  rcy)zyReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrrrrrs rr$)r/ contextlibrtypingrrrrr _devicer __all__strr0 _device_tr:rr r rrr@r<rrrrrrrrrrVs .''   '3T) * 0d0 d  T       h% 9  +*+<!6!4! c  y T r