Vh*nddlZddlZddlmZddlmZddlmZddl m Z m Z m Z  ddl ZedefdZdefdZd eddfd Zdefd Zd eddfd ZdefdZddZeddZGdde Zeej6eeej6e<erdndaerdayday#e$rdZYwxYw)N)contextmanager) lru_cache)Any)__allow_nonbracketed_mutation ContextProp PropModulereturnctduS)aReturn a bool indicating if opt_einsum is currently available. You must install opt-einsum in order for torch to automatically optimize einsum. To make opt-einsum available, you can install it along with torch: ``pip install torch[opt-einsum]`` or by itself: ``pip install opt-einsum``. If the package is installed, torch will import it automatically and use it accordingly. Use this function to check whether opt-einsum was installed and properly imported by torch. N _opt_einsumR/home/dcms/DCMS/lib/python3.12/site-packages/torch/backends/opt_einsum/__init__.py is_availablers d ""rctS)zNReturn the opt_einsum package if opt_einsum is currently available, else None.r r rrget_opt_einsumrs r_enabledc>ts|rtd|d|ay)Nz5opt_einsum is not available, so setting `enabled` to z will not reap the benefits of calculating an optimal path for einsum. torch.einsum will fall back to contracting from left to right. To enable this optimal path calculation, please install opt-einsum.)r ValueErrorenabled)rs r _set_enabledr#s/ >hCH:N6 6  GrctSN)rr rr _get_enabledr/s Nr _strategyctstd|dtstd|d|dvrtd||ay)Nz6opt_einsum is not available, so setting `strategy` to z will not be meaningful. torch.einsum will bypass path calculation and simply contract from left to right. Please install opt_einsum or unset `strategy`.z6opt_einsum is not enabled, so setting a `strategy` to z will not be meaningful. torch.einsum will bypass path calculation and simply contract from left to right. Please set `enabled` to `True` as well or unset `strategy`.)autogreedyoptimalzH`strategy` must be one of the following: [auto, greedy, optimal] but is )rrrstrategy)rs r _set_strategyr!3sw >DYKP= =  DYKPJ J  55VW`Va b  HrctSr)r r rr _get_strategyr#Hs Orcjttsdntf}| t|| t ||Sr)rrr rr!)rr orig_flagss r set_flagsr&Ls4|~48DJXi rc#Kt5t||}ddd dt5tdddy#1swY*xYw#1swYyxYw#t5tdddw#1swYwxYwxYwwr)rr&)rr r%s rflagsr(Us & (2w1 2# + , # z " # # 22  # # * , # z " # # #s_ B ABA B A BA BAB B& A8/ B8B =BBc\eZdZfdZeeeadae r ee e axZ SxZ S)OptEinsumModulec&t|||yr)super__init__)selfmname __class__s rr-zOptEinsumModule.__init__gs D!rN) __name__ __module__ __qualname__r-rrrrr rr#r! __classcell__)r1s@rr*r*fs1", 5GH~}m<rr*TFr)NN)syswarnings contextlibr functoolsr _lru_cachetypingrtorch.backendsrrr opt_einsumr ImportErrorboolrrrrstrr!r#r&r(r*modulesr2rr r rrrBs %-QQ$   #d #  # 4 D dST*s## =j =( H(=xH H.$e!^6UKsB**B43B4