o /iF@sddlZddlZddlmZddlZdaddZddZedd ZGd d d Z Gd d d ej Z GdddZ dddZ edddZddZddZGddde Zeejeeeje<ddlmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"dS)N)contextmanagerTcCsdadS)NF"__allow_nonbracketed_mutation_flagrrP/home/ubuntu/LTX-2/.venv/lib/python3.10/site-packages/torch/backends/__init__.pydisable_global_flagssrcCst SNrrrrr flags_frozensr ccs"t}dazdVW|adS|aw)NTr)oldrrr__allow_nonbracketed_mutations r c@s$eZdZddZddZddZdS) ContextPropcC||_||_dSr)gettersetter)selfrrrrr__init__% zContextProp.__init__cCs|Sr)r)robjobjtyperrr__get__)szContextProp.__get__cCs&ts ||dStd|jd)Nznot allowed to set zM flags after disable_global_flags; please use flags() context manager instead)r r RuntimeError__name__)rrvalrrr__set__,s  zContextProp.__set__N)r __module__ __qualname__rrrrrrrr $s r cs$eZdZfddZddZZS) PropModulecst|||_dSr)superrmrrname __class__rrr7s  zPropModule.__init__cCs |j|Sr)r__getattribute__)rattrrrr __getattr__;s zPropModule.__getattr__)rrrrr% __classcell__rrr!rr6s rcs,eZdZddZfddZddZZS)_FP32PrecisioncCr rbackendop)rr)r*rrrr@rz_FP32Precision.__init__csF|dkrtj|j|j|dS|dvrt||dStd|)Nfp32_precisionr(Unknown attribute )torch_C_set_fp32_precision_setterr)r*r __setattr__AttributeError)rr valuer!rrr0Ds  z_FP32Precision.__setattr__cCs&|dkr tj|j|jStd|)Nr+r,)r-r._get_fp32_precision_getterr)r*r1)rr rrrr%Ls z_FP32Precision.__getattr__)rrrrr0r%r&rrr!rr'?s r'nonecCs,tjddf}|durtjdd||S)Ngenericall)r-r.r3r/)_fp32_precision orig_flagsrrr set_flagsSsr9ccst t|}Wdn1swYzdVWt t|WddS1s0wYdSt t|Wdw1sIwYwr)r r9)r+r8rrrflagsZs  * r:cfdd}|S)NcstjSr)r-r.r3rr(rrinnerfsz)_get_fp32_precision_getter..innerrr)r*r<rr(rr3er3cr;)Ncstj|Sr)r-r.r/) precisionr(rrr<msz)_set_fp32_precision_setter..innerrr=rr(rr/lr>r/cs2eZdZfddZeeddeddZZS) GenericModulecst||dSr)rrrr!rrrtszGenericModule.__init__r5r6) rrrrr r3r/r+r&rrr!rr@ss   r@)cpucudacudnn cusparseltkleidiaimhamiopenmklmkldnnmpsnnpackopenmp opt_einsum quantized)r4)#systypes contextlibrr-rrr r r ModuleTyperr'r9r:r3r/r@modulesrtorch.backendsrArBrCrDrErFrGrHrIrJrKrLrMrNrrrrs(       D