v0.8.1

New features

Kernel version bounds

get_kernel adds a version argument, which you can use to fetch the latest version compatible with the given version specifier:

activation = kernels.get_kernel("kernels-community/activation", version=">=0.0.3,<0.1")

Version bounds are now also supported when registering layers:

kernel_layer_mapping = {
    "SiluAndMul": {
        "cuda": LayerRepository(
            repo_id="kernels-community/activation",
            version=">=0.0.3,<0.1",
            layer_name="SiluAndMul",
        )
    }
}

Kernel layer locking

Layers can now also use version locks from kernels.lock by using LockedLayerRepository:

kernel_layer_mapping = {
    "SiluAndMul": {
        "cuda": LockedLayerRepository(
            repo_id="kernels-community/activation",
            layer_name="SiluAndMul",
        )
    }
}

See the kernel locking documentation for more information.

What's Changed

• get_kernel: allow Python-style version specifiers by @danieldk in #111
• triton based kernel could also run in xpu by @sywangyi in #112
• Add version support to LayerRepository by @danieldk in #113
• Add support for project-wide locking of layers by @danieldk in #114
• Set version to 0.8.1.dev0 by @danieldk in #115

New Contributors

• @sywangyi made their first contribution in #112

Full Changelog: v0.8.0...v0.8.1