unsloth/unsloth/kernels
Datta Nimmaturi a459558421 Move inputs to right devices. (#2919)
* Move tensors to right devices

* fix multi gpu for non mistral models

* multi GPU RoPE for gemma2

* Finish up multi GPU inference

* Make multiGPU rope a list

* Remove unnecessary transfer to CPU

* Remove unnecessary move to CPU

* Donot move inputs to device yet

will be handled separately in another PR

* Move inputs to appropriate decoder device

* Make device count global variable

* Cleanup RoPE device code

* Fixup num_gpu to device count

* Cleanup device counts

* Use device index for RoPE get_cache

* Donot typecast

* Use tuple instead of list for tensors. Use device index directly

* fixup move to device logic
2025-07-10 04:01:03 -07:00
..
moe MoE kernels AGPLv3 2025-07-06 22:44:35 -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 Move inputs to right devices. (#2919) 2025-07-10 04:01:03 -07:00