unsloth/unsloth/kernels
Lei Zhenyuan 01c5e1a249
[3/N] Enable intel GPU for unsloth (#2620)
* enable intel xpu changes within kernels

* reslove torch.version < 2.6

* change version check to 2.6.0

* resolve comments for torch_gpu_device

* resolve amp fwd comments

* fix typo

* change cuda default logic

* clean this pr

* add HAS_CUDA_STREAM as default False

* split GPU streams to cuda and xpu streams

* add optional
2025-06-24 02:01:28 -07:00
..
moe Docs: Fix typo and improve MoE docstrings (#2784) 2025-06-23 01:09:23 -07:00
__init__.py patch_fast_lora 2024-11-20 03:36:41 -08:00
cross_entropy_loss.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00
fast_lora.py LoRA 2025-03-01 02:52:20 -08:00
flex_attention.py Bug fixes (#1516) 2025-01-07 04:23:14 -08:00
geglu.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00
layernorm.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00
rms_layernorm.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00
rope_embedding.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00
swiglu.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00
utils.py [3/N] Enable intel GPU for unsloth (#2620) 2025-06-24 02:01:28 -07:00