a  þd›ã@s(dZddlZddlZdd„Zdd„ZdS)zP JIT scripting/tracing utils Hacked together by / Copyright 2020 Ross Wightman éNcCs<ttjdƒsJdƒ‚tj d¡tj d¡tj d¡dS)zÞ Set JIT executor to legacy w/ support for op fusion This is hopefully a temporary need in 1.5/1.5.1/1.6 to restore performance due to changes in the JIT exectutor. These API are not supported so could change. Ú_jit_set_profiling_executorzOld JIT behavior doesn't exist!FTN)ÚhasattrÚtorchÚ_CrÚ_jit_set_profiling_modeÚ_jit_override_can_fuse_on_gpu©rrúW/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/utils/jit.pyÚset_jit_legacy s  r cCsr|dkrltj d¡tj d¡tj d¡tj d¡tj d¡ztj d¡WntyfYn0n|dks||dkrÒtj d¡tj d¡tj d¡tj d¡ztj d¡WntyÎYn0nœ|dksä|dkrXdt j d <tj d¡tj d¡tj d¡tj  ¡tj  ¡tj d¡tj d¡tj  d¡tj d¡ndsnJd |›d ƒ‚dS) NÚteTFÚoldÚlegacyZnvfuserZnvfÚ1Z PYTORCH_NVFUSER_DISABLE_FALLBACKzInvalid jit fuser (ú))rrrrZ_jit_override_can_fuse_on_cpurZ_jit_set_texpr_fuser_enabledZ_jit_set_nvfuser_enabledÚ ExceptionÚosÚenvironZ_jit_can_fuse_on_cpuZ_jit_can_fuse_on_gpuZ_jit_set_nvfuser_guard_mode)Zfuserrrr Ú set_jit_fusers>                     r)Ú__doc__rrr rrrrr Ús