U yh@sUddlmZddlmZddlZddlmZdddd gZ e d d dZ ee d d dZ ee e e d dd Zede e dddZdaeeed<ddZdS)) lru_cache)OptionalN)Libraryis_built is_availableis_macos13_or_neweris_macos_or_newer)returncCstjjS)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_CZ_has_mpsr r M/var/www/html/venv/lib/python3.8/site-packages/torch/backends/mps/__init__.pyr scCs tjS)z7Return a bool indicating if MPS is currently available.)r r Z_mps_is_availabler r r rrs)majorminorr cCstj||S)zHReturn a bool indicating whether MPS is running on given MacOS or newer.r r Z_mps_is_on_macos_or_newer)rrr r rr s)rr cCstjd|S)zEReturn a bool indicating whether MPS is running on MacOS 13 or newer. r)rr r rr"s_libcCsXtdkstdk rdSddlm}ddlm}tddatd|d td |d dS) zs