
    Ph                     t    d dl mZ d dlZddlmZ g dZdefdZ	edefd       Z
edd	edefd
       Zdad Zy)    )	lru_cacheN   )Library)is_builtis_availableis_macos13_or_newerreturnc                  6    t         j                  j                  S )zReturn whether PyTorch is built with MPS support.

    Note that this doesn't necessarily mean MPS is available; just that
    if this PyTorch binary were run a machine with working MPS drivers
    and devices, we would be able to use it.
    )torch_C_has_mps     fC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/backends/mps/__init__.pyr   r   	   s     88r   c                  >    t         j                  j                         S )z7Return a bool indicating if MPS is currently available.)r   r   _mps_is_availabler   r   r   r   r      s     88%%''r   minorc                 @    t         j                  j                  |       S )zEReturn a bool indicating whether MPS is running on MacOS 13 or newer.)r   r   _mps_is_on_macos_13_or_newer)r   s    r   r   r      s     880077r   c                      t               du st        yddlm}  ddlm}m} t        dd      at        j                  d|d	       t        j                  d
|d	       t        j                  d| d	       y)z<Register prims as implementation of var_mean and group_norm.FNr   )native_group_norm_backward)native_group_normvar_meanatenIMPLzvar_mean.correctionMPSr   r   )	r   _lib_decomp.decompositionsr   _refsr   r   _Libraryimpl)_native_group_norm_backward_native_group_norm	_var_means      r   _initr%   "   sb     zUd. XFF#DII#Y6II!#5u=II*,GOr   )r   )	functoolsr   
_lru_cacher   libraryr   r    __all__boolr   r   intr   r   r%   r   r   r   <module>r,      si    -  *
=$  (d ( (
 8s 84 8 8
 Pr   