mirror of
https://github.com/huggingface/kernels.git
synced 2025-10-20 21:10:02 +08:00
d383fdd4b4
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>
7 lines
287 B
INI
7 lines
287 B
INI
[pytest]
|
|
markers =
|
|
cuda_only: marks tests that should only hosts with CUDA GPUs
|
|
rocm_only: marks tests that should only run on hosts with ROCm GPUs
|
|
darwin_only: marks tests that should only run on macOS
|
|
xpu_only: marks tests that should only run on hosts with Intel XPUs
|