Skip to content

Commit e39eac0

Browse files
authored
up (#159)
1 parent b0c431f commit e39eac0

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

docs/source/kernel-requirements.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,16 @@ have dynamic library dependencies outside:
4646
- Torch;
4747
- CUDA/ROCm libraries installed as dependencies of Torch.
4848

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+
4959
### Linux
5060

5161
- Use [ABI3/Limited API](https://docs.python.org/3/c-api/stable.html#stable-application-binary-interface)

0 commit comments

Comments
 (0)