# {py:mod}`evox.triton_kernels.op_register`

```{py:module} evox.triton_kernels.op_register
```

```{autodoc2-docstring} evox.triton_kernels.op_register
:allowtitles:
```

## Module Contents

### Functions

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`register_triton_op <evox.triton_kernels.op_register.register_triton_op>`
  - ```{autodoc2-docstring} evox.triton_kernels.op_register.register_triton_op
    :summary:
    ```
````

### API

````{py:function} register_triton_op(*, fake_fn, triton_fn, vmap_fn=None, fake_vmap_fn=None, vmap_wrap_inputs=None, vmap_out_dims=0, max_vmap_level=None, name=None, mutates_args=(), device_types=None, triton_device_types: str | typing.Sequence[str] | None = None, schema=None)
:canonical: evox.triton_kernels.op_register.register_triton_op

```{autodoc2-docstring} evox.triton_kernels.op_register.register_triton_op
```
````