^jaddlZddlZddlmZddlmZddlmZmZeje Z GddeZ Gdd Z e Zy) N)Protocol)Version) _OpCondFn _OpImplFncleZdZdZdefdZdedzfdZddZdddd e d e d e d e dzd e dededdfdZ y)DSLModuleProtocolz*Complete interface for DSL utility modulesreturncyNselfs e/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_native/dsl_registry.pyruntime_availablez#DSLModuleProtocol.runtime_availableNcyr r rs rruntime_versionz!DSLModuleProtocol.runtime_versionrrcyr r rs rderegister_op_overridesz)DSLModuleProtocol.deregister_op_overridesrrF)allow_multiple_overrideunconditional_override lib_symbol op_symbol dispatch_keycondimplrrcyr r )rrrrrrrrs rregister_op_overridez&DSLModuleProtocol.register_op_overridesr)r N) __name__ __module__ __qualname____doc__boolrrrrstrrrr r rrr r s|4,4,4442).',     $   "& !%   rr ceZdZdZdZdeddfdZdededdfdZe jd ede fd Z e jd ede dzfd Ze jdeed ffd Ze jdeed ffdZdededzfdZy) DSLRegistryzIRegistry for DSL modules - calls their existing API functions dynamicallyci|_yr ) _dsl_modulesrs r__init__zDSLRegistry.__init__'s :<rnamer Nct|ts!tdt|j|j s t dy)zValidate DSL name at runtimezDSL name must be string, got z&DSL name cannot be empty or whitespaceN) isinstancer& TypeErrortyper!strip ValueErrorrr,s r_validate_dsl_namezDSLRegistry._validate_dsl_name*sD$$;DJzDSLRegistry.list_available_dslssG %% 'D$$T*  & 'YrcHt|jjS)z/Get all registered DSL names (available or not))rRr*keysrs rr?zDSLRegistry.list_all_dslssT&&++-..rc8|jj|S)zGet a registered DSL module by name. Args: name: Name of the DSL to retrieve. Returns: The DSL module if registered, None otherwise. )r*rJr3s rget_dsl_modulezDSLRegistry.get_dsl_modules  $$T**r)r!r"r#r$r+r&r4r rE functoolscacher%r;rr=rRr>r?rWr rrr(r($sS=GsGtG0:0:2C0:0:d__    __  $  __ U38_  __/uS#X// +3 ++r`sL3*g!*r+r+l} r