U yhD@sTddlmZddlZddlmZmZmZdddgZddZddZ ed d dZ dS) )contextmanagerN)__allow_nonbracketed_mutation ContextProp PropModule is_availableflags set_flagscCstS)z4Return whether PyTorch is built with NNPACK support.)torchZ_nnpack_availabler r P/var/www/html/venv/lib/python3.8/site-packages/torch/backends/nnpack/__init__.pyr scCstjf}tj||S)z!Set if nnpack is enabled globally)r Z_CZ_get_nnpack_enabledZ_set_nnpack_enabled)Z_enabled orig_flagsr r r rs  FccsJtt|}W5QRXz dVW5tt|dW5QRXXdS)z9Context manager for setting if nnpack is enabled globallyrN)rr)Zenabledr r r r rs  )F) contextlibrr Ztorch.backendsrrr__all__rrrr r r r s