frozenleaves/kernels
1
0
Fork 0
mirror of https://github.com/huggingface/kernels.git synced 2025-10-20 20:56:31 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
a8a6564fa7bdb8992aad5faf31da6b19896bb9bc
kernels/tests
History
Ákos Hadnagy 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
..
kernel_locking
Make the forward pass torch.compile compatible (#87)
2025-06-03 15:06:02 +02:00
layer_locking
Add support for project-wide locking of layers (#114)
2025-07-23 09:37:05 +02:00
conftest.py
Add ROCm device discovery (#122)
2025-08-01 16:09:45 +02:00
test_basic.py
Test examples in docstrings using mktestdocs (#118)
2025-07-28 17:31:34 +02:00
test_benchmarks.py
Test examples in docstrings using mktestdocs (#118)
2025-07-28 17:31:34 +02:00
test_doctest.py
Test examples in docstrings using mktestdocs (#118)
2025-07-28 17:31:34 +02:00
test_interval_tree.py
Support registering layers with a range of CUDA capabilities (#106)
2025-07-14 16:59:21 +02:00
test_kernel_locking.py
Test examples in docstrings using mktestdocs (#118)
2025-07-28 17:31:34 +02:00
test_layer.py
Add ROCm device discovery (#122)
2025-08-01 16:09:45 +02:00