U ¯Ê¹j›ã@s(dZddlZddlZdd„Zdd„ZdS)zP JIT scripting/tracing utils Hacked together by / Copyright 2020 Ross Wightman éNcCs<ttjdƒstdƒ‚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Ú_CÚAssertionErrorrÚ_jit_set_profiling_modeÚ_jit_override_can_fuse_on_gpu©r r úM/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/utils/jit.pyÚset_jit_legacy s  r cCsv|dkrntj d¡tj d¡tj d¡tj d¡tj d¡ztj d¡Wntk rhYnXn|dks~|dkrÖtj d¡tj d¡tj d¡tj d¡ztj d¡Wntk rÒYnXnœ|dksè|dkr\dt j d <tj d¡tj d¡tj d¡tj  ¡tj  ¡tj d¡tj d¡tj  d¡tj d¡ndsrtd |›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_moder)Zfuserr r r Ú set_jit_fusers>                   r)Ú__doc__rrr rr r r r Ús