o I hE@sRddlmZddlZddlmZmZmZgdZddZddZ ed d d Z dS) )contextmanagerN)__allow_nonbracketed_mutation ContextProp PropModule) is_availableflags set_flagscCstS)z4Return whether PyTorch is built with NNPACK support.)torch_nnpack_availabler r R/var/www/vscode/kcb/lib/python3.10/site-packages/torch/backends/nnpack/__init__.pyr srcCstjf}tj||S)z!Set if nnpack is enabled globally)r _C_get_nnpack_enabled_set_nnpack_enabled)_enabled orig_flagsr r r rs  rFccst t|}Wdn1swYz dVWtt|dWddS1s2wYdStt|dWdw1sMwYw)z9Context manager for setting if nnpack is enabled globallyNr)rr)enabledrr r r rs *r)F) contextlibrr torch.backendsrrr__all__rrrr r r r s