PhddlZddlZddlZ ddlZdZddlmZgdZ GddejjjZ dZ d ddd Zd Zy#e$rdZYJwxYw) NT)Any)autocast custom_fwd custom_bwdceZdZdZdej dfdedejdeffd ZfdZ de d e d e ffd Z fd Z xZ S) rzSee :class:`torch.autocast`. ``torch.cuda.amp.autocast(args...)`` is equivalent to ``torch.autocast("cuda", args...)`` Tenableddtype cache_enabledctjjr||_d|_||_yt |d|||y)Ncuda)rr r )torch _jit_internal is_scripting_enableddevice fast_dtypesuper__init__)selfrr r __class__s gC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/cuda/amp/autocast_mode.pyrzautocast.__init__sJ    + + -#DM DK#DO   G5   c`tjjr|St|SN)r rrr __enter__)rrs rrzautocast.__enter__&s(    + + -Kw ""rexc_typeexc_valexc_tbcdtjjryt||||Sr)r rrr__exit__)rrrrrs rr zautocast.__exit__,s,    + + - w'6::rcbtjjr|St||Sr)r rrr__call__)rfuncrs rr"zautocast.__call__1s*    + + -Kw%%r)__name__ __module__ __qualname____doc__r float16boolr rrrr r" __classcell__)rs@rrrse"]]"    {{    # ;;s;C; &&rrc t|tjrQ|jxr*|jxr|j tj u}|r|jS|St|ttfr|Strt|tjr|St|tjjr8|j!Dcic]\}}t#|t#|c}}St|tjj$r4fd|D}t|t&t(frt+||S|S|Scc}}w)Nc36K|]}t|ywr)_cast).0vr s r z_cast..Hs3UE!UOUs) isinstancer Tensoris_floating_pointis_cudar float64tostrbytes HAS_NUMPYnpndarray collectionsabcMappingitemsr-Iterablelisttupletype)valuer is_eligiblekr/iterables ` rr-r-9s%&  # # % 3  3EMM1  #.uxx858 EC< ( z%4 E;??22 3=B[[]K]TQaq%0]KK E;??33 43U3 edE] +4;x( (O Ls!E cast_inputsctjtStjfd}|S)aa Create a helper decorator for ``forward`` methods of custom autograd functions. Autograd functions are subclasses of :class:`torch.autograd.Function`. See the :ref:`example page` for more detail. Args: cast_inputs (:class:`torch.dtype` or None, optional, default=None): If not ``None``, when ``forward`` runs in an autocast-enabled region, casts incoming floating-point CUDA Tensors to the target dtype (non-floating-point Tensors are not affected), then executes ``forward`` with autocast disabled. If ``None``, ``forward``'s internal ops execute with the current autocast state. .. note:: If the decorated ``forward`` is called outside an autocast-enabled region, :func:`custom_fwd` is a no-op and ``cast_inputs`` has no effect. rHc Ttj|d_$tj|d_|i|Stj}d|d_|r2t d5t |it |cdddS|i|S#1swYyxYw)NrF)r)r get_autocast_gpu_dtype_dtypeis_autocast_enabled_fwd_used_autocastrr-)argskwargsautocast_contextrIfwds r decorate_fwdz custom_fwd..decorate_fwdns557Q  ).)B)B)DDG &'' '$88: ).DG &e,dK 8WE&+` for more detail. ct|dj|dj5|i|cdddS#1swYyxYw)Nr)rr )rrOrM)rPrQbwds r decorate_bwdz custom_bwd..decorate_bwds9 d1g88Q O''P O Os ;A)rUrW)rZr[s` rrrs(__S(( rr)r<rUr numpyr:r9ModuleNotFoundErrortypingr__all__amp autocast_moderr-rrrrrcsq I 2#&uyy&&//#&P@$$T s BsAAA