hC SSKJr SSKrSSKJr SSKJr SSKJr SSK J r J r J r J r Jr SSKJr \ "S5r"S S \5r"S S \\5rSSSSSSSSS .SSjjjrg)) annotationsN ASTSource)Language) JITFunction)TypeVarOptionalCallableIterableUnion)irTc6^\rSrSrSSU4SjjjrSrSrU=r$)GluonASTSource c^>[TU]XX45 [RUlSUlg)Nttgir)super__init__rGLUONlanguageext)selffn signature constexprsattrs __class__s \/var/www/fran/franai/venv/lib/python3.13/site-packages/triton/experimental/gluon/_runtime.pyrGluonASTSource.__init__s$  :  c SSKJn SSKJn [R "U5nUR 5n[RRRR5n U"U 5n U RU5n URSURU 55 URSURUR 55 URSURUR"55 URSURS55 UR$b+URS URUR$55 U"UR&XXX8S 9nU$) Nr) make_backend) ast_to_ttirz ttg.targetz ttg.num-warpsz ttg.num-ctaszttg.threads-per-warp z ttg.maxnreg)contextoptions codegen_fns module_mapmodule)triton.compiler.compilerr#triton.compiler.code_generatorr$r builder create_moduletritonruntimedriveractiveget_current_targetget_target_nameset_attrget_string_attrget_int32_attr num_warpsnum_ctasmaxnregr) rr'r(r)r&r#r$r-r*targetbackends rmake_irGluonASTSource.make_irs9>**W%&&(&&--@@Bv&((1 g&=&=f&EF)?)?@Q@Q)RS(>(>w?O?O(PQ.0F0Fr0JK ?? & OOM7+A+A'//+R STWWdW(2C r!)rr)NN)returnNone)__name__ __module__ __qualname____firstlineno__rr=__static_attributes__ __classcell__rs@rrr s r!rc.^\rSrSrU4SjrSrSrU=r$)GluonJITFunction*c:>[TU]5n[UlU$N)r create_binderrr)rresultrs rrMGluonJITFunction.create_binder,s&(' r!cg)NT)rs ris_gluonGluonJITFunction.is_gluon1sr!r)rArBrCrDrMrRrErFrGs@rrIrI*s r!rI)versionreprlaunch_metadatado_not_specializedo_not_specialize_on_alignmentdebugnoinlinecF^^^^^^^SUUUUUUU4SjjnUbU"U5$U$)a  Decorator for JIT-compiling a function using the Triton compiler. :note: When a jit'd function is called, arguments are implicitly converted to pointers if they have a :code:`.data_ptr()` method and a `.dtype` attribute. :note: This function will be compiled and run on the GPU. It will only have access to: * python primitives, * builtins within the triton package, * arguments to this function, * other jit'd functions :param fn: the function to be jit-compiled :type fn: Callable c H>[U5(de[UTTTTTTTS9$)N)rTrWrXrYrZrUrV)callablerI)rrYrWrXrVrZrUrTs r decoratorjit..decoratorRs6||| /+I+  r!)rrr?zJITFunction[T]rQ) rrTrUrVrWrXrYrZr^s ``````` rjitr`5s&:    ~}r!rL)rz Optional[T]rUOptional[Callable]rVrarWOptional[Iterable[int | str]]rXrbrYOptional[bool]rZrcr?z9Union[GluonJITFunction[T], Callable[[T], JITFunction[T]]]) __future__rr/r+rtriton.backends.compilerrtriton.runtime.jitrtypingrr r r r triton._C.libtritonr rrrIr`rQr!rris" .-*??" CLY<{1~. #*.7;DH #..  . ( . 5 .%B. ..?.r!