o Y۷i @sZdZddlZddlmZmZddlmZmZGdddZej Z GdddZ d d Z dS) z Target Options N)configutils) TargetConfigOptionc@s:eZdZdZGdddZddZeddZdd Zd S) TargetOptionszTarget options maps user options from decorators to the ``numba.core.compiler.Flags`` used by lowering and target context. c@seZdZddfddZdS)zTargetOptions.MappingcCs|SN)xrrH/home/ubuntu/vllm_env/lib/python3.10/site-packages/numba/core/options.pyszTargetOptions.Mapping.cCs||_||_dSr) flag_nameapply)selfr r rrr __init__s zTargetOptions.Mapping.__init__N)__name__ __module__ __qualname__rrrrr MappingsrcCsdS)zSubclasses can override this method to make target specific customizations of default flags. Parameters ---------- flags : Flags options : dict Nr)rflagsoptionsrrr finalizes zTargetOptions.finalizecCs"|}|||||||S)zParse target options defined in ``options`` and set ``flags`` accordingly. Parameters ---------- flags : Flags options : dict )_applyr)clsrroptrrr parse_as_flagss  zTargetOptions.parse_as_flagsc Csi}t|}t|D]}t||}t||jr|||<q t}|D]\}}||vr>|||}t||j || |q#t||} | rUd| d| } t | dS)NzUnrecognized options: z. Known options are ) typedirgetattr isinstancersetitemsr setattrr addkeysKeyError) rrrmappingsrkvusedmappingunusedmrrr r-s*     zTargetOptions._applyN) rrr__doc__rr classmethodrrrrrr r s  rc@seZdZdZedejZedZedZ edZ edZ edZ edZ ed Zed Zed Zed Zed ZedZedZedZedZedZedZdS)DefaultOptionszJDefines how user-level target options are mapped to the target flags. enable_pyobjectforce_pyobjectenable_loopliftnrt debuginfo boundscheck release_gil writable_args no_rewritesno_cpython_wrapperno_cfunc_wrapper auto_parallelfastmath error_modelinline forceinlinedbg_extend_lifetimes dbg_optnoneN)rrrr,_mappingoperatornot_nopythonforceobjlooplift_nrtdebugr4nogilr6r7r8r9parallelr;r<r=r>_dbg_extend_lifetimes _dbg_optnonerrrr r.Hs(  r.cGsdd|D}tdd|S)zReturns a mixin class with a subset of the options Parameters ---------- *args : str Option names to include. cSsi|]}|tt|qSr)rr.).0r&rrr jsz+include_default_options.. OptionMixinsr)r)argsglbsrrr include_default_optionsbs rR) r,rB numba.corerrnumba.core.targetconfigrrrrrAr.rRrrrr s;