o ZhE@sRddlmZddlZddlmZmZmZgdZddZddZ ed d d Z dS) )contextmanagerN)__allow_nonbracketed_mutation ContextProp PropModule) is_availableflags set_flagscCstS)z4Return whether PyTorch is built with NNPACK support.)torchZ_nnpack_availabler r M/var/www/auris/lib/python3.10/site-packages/torch/backends/nnpack/__init__.pyr srcCstjf}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  rFccst t|}Wdn1swYz dVWtt|dWddS1s2wYdStt|dWdw1sMwYw)z9Context manager for setting if nnpack is enabled globallyNr)rr)enabledr r r r rs *r)F) contextlibrr Ztorch.backendsrrr__all__rrrr r r r s