U Kºcy ã@s.ddlmZddlZddlZddlmZddlmZddl m Z m Z m Z z ddl ZWnek rldZYnXeƒedœdd„ƒZedœd d „Zedd œd d „Zedœdd„Zeddœdd„Zedœdd„Zddd„Zeddd„ƒZGdd„de ƒZeejeeƒeje<eƒrdndaeƒr&dndadS) é)ÚAnyN)Ú lru_cache)Úcontextmanager)Ú ContextPropÚ PropModuleÚ__allow_nonbracketed_mutation)ÚreturncCstdk S)z?Returns a bool indicating if opt_einsum is currently available.N©Ú _opt_einsum©r r úF/tmp/pip-unpacked-wheel-gikjz4vx/torch/backends/opt_einsum/__init__.pyÚ is_availablesr cCstS)zOReturns the opt_einsum package if opt_einsum is currently available, else None.r r r r r Úget_opt_einsumsr)Ú_enabledrcCs"tƒs|rtd|›dƒ‚|adS)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 Ú ValueErrorÚenabled)rr r r Ú _set_enableds rcCstS©N)rr r r r Ú _get_enabled#sr)Ú _strategyrcCsHtƒstd|›dƒ‚ts*td|›dƒ‚|dkr@td|›ƒ‚|adS)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`.)ÚautoZgreedyZoptimalzH`strategy` must be one of the following: [auto, greedy, optimal] but is )r rrÚstrategy)rr r r Ú _set_strategy'srcCstSr)rr r r r Ú _get_strategy6srcCs6ttƒs dntf}|dk r"t|ƒ|dk r2t|ƒ|Sr)rr rrr)rrÚ orig_flagsr r r Ú set_flags:s rccsHtƒt||ƒ}W5QRXz dVW5tƒt|ŽW5QRXXdSr)rr)rrrr r r ÚflagsCs  rcs:eZdZ‡fdd„Zeeeƒadae ƒr2ee e ƒa‡Z S)ÚOptEinsumModulecstt|ƒ ||¡dSr)ÚsuperrÚ__init__)ÚselfÚmÚname©Ú __class__r r rTszOptEinsumModule.__init__N) Ú__name__Ú __module__Ú __qualname__rrrrrrr rrÚ __classcell__r r r#r rSs   rTFr)NN)NN)ÚtypingrÚwarningsÚsysÚ functoolsrZ _lru_cacheÚ contextlibrZtorch.backendsrrrZ opt_einsumr Ú ImportErrorÚboolr rrrÚstrrrrrrÚmodulesr%rrr r r r Ús.