triton.experimental.gluon.GluonJITFunction

class triton.experimental.gluon.GluonJITFunction(self, fn, version=None, do_not_specialize=None, do_not_specialize_on_alignment=None, debug=None, noinline=None, repr=None, launch_metadata=None)

__init__(self, fn, version=None, do_not_specialize=None, do_not_specialize_on_alignment=None, debug=None, noinline=None, repr=None, launch_metadata=None)

Methods

__init__(self, fn[, version, ...])
add_pre_run_hook(self, hook)	Add a hook that will be executed prior to the execution of run function with args and kwargs passed into the kernel
create_binder(self)	Precompute as much as possible.
get_capture_scope(self)
is_gluon(self)
parse(self)
preload(self, specialization_data)
repr(self, _)
run(self, *args, grid, warmup, **kwargs)
warmup(self, *args, grid, **kwargs)

Attributes

cache_key
src
type