3j+SSKrSSKrSSKJrJr SSKJr SSKJr SSK J r SSK r SSK J r JrJr SSKJrJrJr \R("S 5r"S S 5r\"5rg) N)CallableMapping)contextmanager)cached_property)Any)FunctionConstraintsValidationResultvalidate_function_call)BackendNotFoundErrorBackendNotImplementedErrorNoCapableBackendErrorzcomfy_kitchen.dispatchc |\rSrSrSr\S\\\4S-4Sj5rS\ S\ \ \ 44Sjr S \ S \ S\ S-4S jr S\ S \ 4S jrS \ 4SjrS \ 4SjrS\\ 4SjrS \ S\4SjrS\ 4SjrS \ S \ S\\ \4S\4SjrSS \ S\\ \4S-S\ 4SjjrSS \ S\ S-S\\ \4S-S\4Sjjr\S \ 4Sj5rSrg)BackendRegistryc0Ul0Ul0Ul/SQUl[ 5Ul0Ul[R"5Ul [R"5Ul g)N)cudatritoneager) _backends _capabilities _constraints _priorityset _disabled _unavailable threadingLock_locklocal _thread_local)selfs P/home/wildlama/miniconda3/lib/python3.13/site-packages/comfy_kitchen/registry.py__init__BackendRegistry.__init__sN4^^% &__.returnNc[RR5(aT[RR[RR 55nUR UR 4$g![a gf=fN)torchr is_availableget_device_propertiescurrent_devicemajorminor Exception)r"propss r#_compute_capability#BackendRegistry._compute_capabilitysg :: " " $ $  889R9R9TU U[[11  sAA99 BBname capabilitiesc@UR X RU'[UR55URU'UR 5HupEXPR X4'M URRUS5 SSS5 g!,(df  g=f)zRegister a backend with its capabilities and constraints. Args: name: Backend name (e.g., "cuda", "eager", "triton") module: The backend module containing implementations capabilities: Dict mapping function names to FunctionConstraints N) rrrkeysritemsrrpop)r"r4moduler5 func_name constraintss r#registerBackendRegistry.register)s|ZZ#)NN4 '*<+<+<+>'?D  t $*6*<*<*>& 7B!!4"34+?    ! !$ - ZZs A9B B backend_namer;c:URRX45$)z5Get constraints for a specific backend/function pair.)rget)r"r?r;s r#get_constraintsBackendRegistry.get_constraints=s  $$l%>??r&reasonclUR X RU'SSS5 g!,(df  g=f)zMark a backend as unavailable with a reason. Args: name: Backend name reason: Why the backend is unavailable (e.g., "ImportError: ...") N)rr)r"r4rDs r#mark_unavailable BackendRegistry.mark_unavailableCs#ZZ&,  d #ZZs% 3cUR URRU5 SSS5 g!,(df  g=f)zPManually disable a backend. Args: backend_name: Name of backend to disable N)rraddr"r?s r#disableBackendRegistry.disableMs' ZZ NN  | ,ZZ 2 AcUR URRU5 SSS5 g!,(df  g=f)z\Re-enable a previously disabled backend. Args: backend_name: Name of backend to enable N)rrdiscardrJs r#enableBackendRegistry.enableVs' ZZ NN " "< 0ZZrM priority_listcpUR [U5UlSSS5 g!,(df  g=f)zhSet backend selection priority order. Args: priority_list: List of backend names in priority order N)rlistr)r"rRs r# set_priorityBackendRegistry.set_priority_s! ZZ!-0DNZZs' 5cHXR;=(a XR;$)zCheck if backend is available and not disabled. Args: backend_name: Name of backend to check Returns: True if backend is registered and not disabled )rrrJs r#r+BackendRegistry.is_availablehs~~-T,nn2TTr&c 0n[UR5[URR55-[URR55-nUH`nX0R;X0R ;URR U5[URR U/55S.X'Mb U$)a*Return dict of all backends with their status. Returns: Dict mapping backend names to status info: { "backend_name": { "available": bool, "disabled": bool, "unavailable_reason": str or None, "capabilities": list[str] } } ) availabledisabledunavailable_reasonr5) rrrr7rrrAsortedr)r"resultall_backend_namesr4s r# list_backendsBackendRegistry.list_backendsss  #dnn&9&9&;"< >LL!H(^#O(.(;(;'U%V" DhZ!NNL$$\22 2 2 6 6|SU KK77 QWX~~060C0C/DBvG\G\F]-^H* LL5| O +$I88r&backendc8U(aX R;a%[X RRUS55eX R;a [US5eXR RU[ 55;a [X!5eU(aIURX!U5nUR(d&[XURSUR305e[URUU5$URX5n[URUU5$)aGet the best implementation for a function. Args: func_name: Name of the function to get backend: Explicit backend to use, or None for auto-select kwargs: Kwargs for constraint validation (empty/None skips validation) Returns: The function implementation Raises: BackendNotFoundError: If explicit backend is not available BackendNotImplementedError: If explicit backend doesn't implement function NoCapableBackendError: If no backend can handle the call not registeredr[ro)rr rrArrrr rkrrrrurvrqrz)r"r;r|rbr^selected_backends r#get_implementation"BackendRegistry.get_implementations* nn,*74E4E4I4I'Sc4dee..(*7J?? 2 2 6 6w FF0DD77FS~~/ H[H[G\\^_e_t_t^uEv;wxx4>>'2I> >33IFt~~&67CCr&c## URU5(dPXR;aURUn[X5eXR;a [US5e[US5e[ UR SS5nXR lSv Uc[UR S5 gX0R lg!Uc[UR S5 fX0R lf=f7f)zContext manager to temporarily use a specific backend. Args: backend_name: Name of backend to use in this context Example: with registry.use_backend("eager"): result = some_function() r[r~rnN)r+rr rrqr!rndelattr)r"r?rDpreviouss r# use_backendBackendRegistry.use_backends  ..000**<8*<@@/*<DD*<9IJJ4--/A4H.:+ ? **,>?6>""3**,>?6>""3sBC/C+C/,C,,C/)rrrrrrr!rr))NN) __name__ __module__ __qualname____firstlineno__r$rtupleintr2strdictr r=rBrFrKrPrTrUrpr+r`rrr rkrzrrrr__static_attributes__r&r#rrs/U38_t%;..3 334 .(@@,/@ t #@ -S-#--C-1311$s)1 U U Ut:" " " S! "  " N,05959S!D(59 59t#+/ #D#Dt#DS!D( #D  #DJ???r&r)loggingrcollections.abcrr contextlibr functoolsrtypingrr*r<r r r exceptionsr r r getLoggerrsrregistryrr&r#rsT-%% VV   3 4\?\?~  r&