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/kernel-requirements.md
+10Lines changed: 10 additions & 0 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -46,6 +46,16 @@ have dynamic library dependencies outside:
46
46
- Torch;
47
47
- CUDA/ROCm libraries installed as dependencies of Torch.
48
48
49
+
## Compatibility with torch.compile
50
+
51
+
The Kernel Hub also encourages to write the kernels in a `torch.compile`
52
+
compliant way. This helps to ensure that the kernels are compatible with
53
+
`torch.compile` without introducing any graph breaks and triggering
54
+
recompilation which can limit the benefits of compilation.
55
+
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
57
+
recompilation triggers during `torch.compile`.
58
+
49
59
### Linux
50
60
51
61
- Use [ABI3/Limited API](https://docs.python.org/3/c-api/stable.html#stable-application-binary-interface)
0 commit comments