o j9:j.@sfddlZddlZddlmZddlmZddlmZee Z GdddeZ Gdd d Z e Z dS) N)Protocol)Version)_OpFnc@sheZdZdZdefddZdedBfddZddd Zd d d d e d e de de dededdfddZ dS)DSLModuleProtocolz*Complete interface for DSL utility modulesreturncCdSNselfr r a/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/_native/dsl_registry.pyruntime_availablez#DSLModuleProtocol.runtime_availableNcCrr r r r r r runtime_versionrz!DSLModuleProtocol.runtime_versioncCrr r r r r r deregister_op_overridesrz)DSLModuleProtocol.deregister_op_overridesF)allow_multiple_overrideunconditional_override lib_symbol op_symbol dispatch_keyimplrrcCrr r )r rrrrrrr r r register_op_overrides z&DSLModuleProtocol.register_op_override)rN) __name__ __module__ __qualname____doc__boolrrrrstrrrr r r r rs,   rc@seZdZdZddZdeddfddZded eddfd d Ze j d ede fd dZ e j d ede dBfddZe j deedffddZe j deedffddZdededBfddZdS) DSLRegistryzIRegistry for DSL modules - calls their existing API functions dynamicallycCs i|_dSr ) _dsl_modulesr r r r __init__&s zDSLRegistry.__init__namerNcCs2t|tstdt|j|stddS)zValidate DSL name at runtimezDSL name must be string, got z&DSL name cannot be empty or whitespaceN) isinstancer TypeErrortyperstrip ValueErrorr r"r r r _validate_dsl_name)s zDSLRegistry._validate_dsl_name dsl_modulecs||gd}fdd|D}|rtd|d|||jvr:|j|}|ur4td|dStd||j|<|j|j|j |j t d |dS) z-Register a DSL module with required interface)rrrrcsg|] }t|s|qSr )hasattr).0methodr*r r =s  z,DSLRegistry.register_dsl..z DSL module 'z' missing required methods: z'DSL '%s' re-registered with same moduleNzPDSL '%s' re-registered with different module object (possibly from test imports)zSuccessfully registered DSL: %s) r)r$r logdebugwarningis_dsl_available cache_clearget_dsl_versionlist_available_dsls list_all_dslsinfo)r r"r*required_methodsmissing_methodsexisting_moduler r.r register_dsl1s6         zDSLRegistry.register_dsldsl_namecCsf|j|}|dur dSz|WSty#tjd|ddYdSty2td|YdSw)z)r r@rrBr0r1rDr r r r5rs   zDSLRegistry.get_dsl_version.cCs,g}|jD] }||r||qt|S)z%Get names of currently available DSLs)r r3appendtuple)r availabler"r r r r6~s    zDSLRegistry.list_available_dslscCst|jS)z/Get all registered DSL names (available or not))rFr keysr r r r r7szDSLRegistry.list_all_dslscCs |j|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 r@r(r r r get_dsl_modules zDSLRegistry.get_dsl_module)rrrrr!rr)rr< functoolscacherr3rr5rFr6r7rIr r r r r#s2 r)rJloggingtypingrtorch._vendor.packaging.versionrregistryr getLoggerrr0rr dsl_registryr r r r s     v