o Iñ hjã@s¨UddlmZddlmZmZddlZddlmZ gd¢Z de fdd„Z ede fd d „ƒZ ed ed ede fd d„ƒZedd ede fdd„ƒZdaee ed<dd„ZdS)é)Ú lru_cache)ÚOptionalÚ TYPE_CHECKINGN)ÚLibrary)Úis_builtÚ is_availableÚis_macos13_or_newerÚis_macos_or_newerÚreturncCstjjS)zóReturn 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©rrúO/var/www/vscode/kcb/lib/python3.10/site-packages/torch/backends/mps/__init__.pyr srcCs tj ¡S)z7Return a bool indicating if MPS is currently available.)r r Ú_mps_is_availablerrrrrs rÚmajorÚminorcCstj ||¡S)zHReturn a bool indicating whether MPS is running on given MacOS or newer.©r r Ú_mps_is_on_macos_or_newer)rrrrrr ór cCstj d|¡S)zEReturn a bool indicating whether MPS is running on MacOS 13 or newer.é r)rrrrr"rrÚ_libcCsTtdustƒs dSddlm}ddlm}tddƒat d|d¡t d |d¡dS) zs