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
shows how this can be done. This compatibility directory is
28
+
automatically created by `kernel-builder`.
23
29
24
30
## Build variants
25
31
@@ -28,6 +34,17 @@ architecture (e.g. x86_64). For compliance with a compute framework and
28
34
architecture combination, all the variants from the [build variant list](https://github.com/huggingface/kernel-builder/blob/main/docs/build-variants.md)
29
35
must be available for that combination.
30
36
37
+
## Kernel metadata
38
+
39
+
The build variant directory can optionally contain a `metadata.json` file.
40
+
Currently the only purpose of the metadata is to specify the kernel python dependencies, for example:
41
+
42
+
```json
43
+
{ "python-depends": ["nvidia-cutlass-dsl"] }
44
+
```
45
+
46
+
The following dependencies are the only ones allowed at this stage: `einops` and `nvidia-cutlass-dsl`
47
+
31
48
## Versioning
32
49
33
50
Kernels are versioned on the Hub using Git tags. Version tags must be of
@@ -50,10 +67,10 @@ have dynamic library dependencies outside:
50
67
51
68
The Kernel Hub also encourages to write the kernels in a `torch.compile`
52
69
compliant way. This helps to ensure that the kernels are compatible with
53
-
`torch.compile` without introducing any graph breaks and triggering
70
+
`torch.compile` without introducing any graph breaks and triggering
54
71
recompilation which can limit the benefits of compilation.
55
72
56
-
[Here](https://github.com/huggingface/kernel-builder/blob/d1ee9bf9301ac8c5199099d90ee1c9d5c789d5ba/examples/relu-backprop-compile/tests/test_relu.py#L162) is a simple test example which checks for graph breaks and
73
+
[Here](https://github.com/huggingface/kernel-builder/blob/d1ee9bf9301ac8c5199099d90ee1c9d5c789d5ba/examples/relu-backprop-compile/tests/test_relu.py#L162) is a simple test example which checks for graph breaks and
0 commit comments