liger-kernels: add device context manager

[vasqu]

It's affecting models that use device map auto (under multi gpu setting) and use_kernels=True, e.g. RMS Norm gets corrupted. But this goes for any triton kernel launch so it is propogated to all of them

For more details, see linkedin/Liger-Kernel#1303

[vasqu]

/kernel-bot build-and-stage

[github-actions]

No kernels provided. Use /kernel-bot <build|security-and-build|build-and-stage|merge-and-upload> <kernel1> [kernel2 ...].

[vasqu]

/kernel-bot build-and-stage liger-kernels

[github-actions]

Build request processed.

Command: /kernel-bot build-and-stage liger-kernels
Mode: build and stage
Target branch: pr-1029
PR head SHA: 83cd58309bce3df43a152e4227bd854a55c40423
Workflows: build.yaml, build-mac.yaml, build-windows.yaml

Dispatched (1):

• liger-kernels (build.yaml): https://github.com/huggingface/kernels-community/actions/runs/29438973379

[vasqu]

/kernel-bot build-and-stage liger-kernels

[github-actions]

Build request processed.

Command: /kernel-bot build-and-stage liger-kernels
Mode: build and stage
Target branch: pr-1029
PR head SHA: 9e9db35482a2cfaf3a5550681820114dcccc4c6f
Workflows: build.yaml, build-mac.yaml, build-windows.yaml

Dispatched (1):

• liger-kernels (build.yaml): https://github.com/huggingface/kernels-community/actions/runs/29498466373

[sayakpaul]

I think it applies to all kernels that don't have this fix as you also pointed out in Slack. But I guess we can treat on a case by case basis?

[vasqu]

Let me hunt them all down as best as I can, I think it's low effort for big impact tbh so no need to delay on case by case imo

[vasqu]

Added it to all triton launches I could find

[vasqu]

/kernel-bot merge-and-upload liger-kernels

[github-actions]

Build request processed.

Command: /kernel-bot merge-and-upload liger-kernels
Mode: merge, build and upload
Target branch: ``
PR head SHA: 95f9074ecceaaddb2753cc7300a297d4eaed5a63
Workflows: `build.yaml, build-mac.yaml, build-windows.yaml`

Merge result: PR is already merged. Continuing with build/upload.

Dispatched (1):

• liger-kernels (build.yaml): https://github.com/huggingface/kernels-community/actions/runs/29746237535