fb8cd99a2c
Add support for NPU kernelize/layers ( #155 )
...
This change add support for Huawei Ascend NPUs. This is #146 with some formatting/typing fixes.
Co-authored-by: zheliuyu <15750543867@163.com >
2025-09-23 10:46:41 +02:00
93e5765611
[tests] turn the kernels upload tests to be staging tests ( #152 )
2025-09-22 18:53:53 +02:00
6c00194680
Improve errors for layer validation ( #145 )
...
* Improve errors for layer validation
Include the repo and layer name as well as the name of the class
that is being compared to (when applicable).
* Remove upload xfail
* Only enable tests that require a token with `--token`
2025-09-16 14:40:54 +02:00
d383fdd4b4
Add support for XPU layer repostories ( #142 )
...
This change adds support for XPU layer repositories, e.g.:
```
kernel_mapping = {
"LigerRMSNorm": {
"xpu": LayerRepository(
repo_id="kernels-community/liger_kernels",
layer_name="LigerRMSNorm",
)
},
}
Co-authored-by: YangKai0616 <kai.yang@intel.com >
2025-09-11 15:51:02 +02:00
a8a6564fa7
Add ROCm device discovery ( #122 )
...
* Add ROCm device discovery
* Ruff
* Address review comments
* Ruff
* Reorg torch import
* Remove redundant import
* Apply suggestions from code review
Co-authored-by: Daniël de Kok <me@danieldk.eu >
* Address review comments
* Validat device type
* Clean diff
* black
* Sync test with repo changes
* black again
---------
Co-authored-by: Daniël de Kok <me@danieldk.eu >
2025-08-01 16:09:45 +02:00
f7490bd0a9
Test examples in docstrings using mktestdocs ( #118 )
...
Also adjust examples so that they are correct.
2025-07-28 17:31:34 +02:00
848c6db87b
Add support for Metal builds ( #89 )
...
* Add support for Metal builds
* Add Metal test, gate tests by OS where necessary
2025-05-30 15:54:28 +02:00
