You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/source/builder/writing-kernels.md
+5-5Lines changed: 5 additions & 5 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -30,9 +30,9 @@ of CMake or setuptools.
30
30
31
31
This page describes the directory layout of a kernel-builder project, the
32
32
format of the `build.toml` file, and some additional Python glue that
33
-
`kernel-builder` provides. We will use a [simple ReLU kernel](../examples/relu)
33
+
`kernel-builder` provides. We will use a [simple ReLU kernel](https://github.com/huggingface/kernels/tree/main/builder/examples/relu)
34
34
as the running example. After reading this page, you may also want to have
35
-
a look at the more realistic [ReLU kernel with backprop and `torch.compile`](../examples/relu-backprop-compile)
35
+
a look at the more realistic [ReLU kernel with backprop and `torch.compile`](https://github.com/huggingface/kernels/tree/main/builder/examples/relu-backprop-compile)
36
36
support.
37
37
38
38
## Kernel project layout
@@ -107,12 +107,12 @@ depends = [ "torch" ]
107
107
108
108
-`maxver`: the maximum CUDA toolkit version (inclusive). This option
109
109
_must not_ be set under normal circumstances, since it can exclude Torch
110
-
build variants that are [required for compliant kernels](https://github.com/huggingface/kernels/blob/main/docs/kernel-requirements.md).
110
+
build variants that are [required for compliant kernels](../kernel-requirements.md).
111
111
This option is provided for kernels that cause compiler errors on
112
112
newer CUDA toolkit versions.
113
113
-`minver`: the minimum required CUDA toolkit version. This option
114
114
_must not_ be set under normal circumstances, since it can exclude Torch
115
-
build variants that are [required for compliant kernels](https://github.com/huggingface/kernels/blob/main/docs/kernel-requirements.md).
115
+
build variants that are [required for compliant kernels](../kernel-requirements.md).
116
116
This option is provided for kernels that require functionality only
117
117
provided by newer CUDA toolkits.
118
118
@@ -145,7 +145,7 @@ The following options can be set for a kernel:
145
145
supported backends are `cpu`, `cuda`, `metal`, `rocm`, and `xpu`.
146
146
**The `cpu` backend is currently experimental and might still change.**
147
147
-`depends` (required): a list of dependencies. The supported dependencies
148
-
are listed in [`deps.nix`](../lib/deps.nix).
148
+
are listed in [`deps.nix`](https://github.com/huggingface/kernels/blob/main/builder/lib/deps.nix).
149
149
-`src` (required): a list of source files and headers.
150
150
-`include` (optional): include directories relative to the project root.
0 commit comments