biondizzle 87b6c9932b Fix CuTeDSL from_dlpack device mismatch inside CUDA graph capture ...

When capturing CUDA graphs on non-default GPUs, torch.cuda.current_device()
may not match the tensor's device. from_dlpack() checks this and fails.
Fix: set the current device to match the tensor's device before from_dlpack.

This enables graph capture on all 8 GPUs, not just cuda:0.

2026-06-03 20:34:24 +00:00

..

_archive

Cleanup Step 2: Archive Lineage P code, fix broken imports

2026-06-02 19:27:07 +00:00

cache

Cleanup Step 2: Archive Lineage P code, fix broken imports

2026-06-02 19:27:07 +00:00

decode

CUDA graph: Fix remaining sync violations from B200 detector run 2

2026-06-03 17:20:34 +00:00

kernels

Fix compressor: do not add positional bias to KV content

2026-06-03 15:52:00 +00:00

layers

CUDA graph: Fix MoE scatter_add_ index dtype + fix second bincount

2026-06-03 17:53:40 +00:00

loader

Cleanup Step 2: Archive Lineage P code, fix broken imports

2026-06-02 19:27:07 +00:00

model

Cleanup Step 2: Archive Lineage P code, fix broken imports

2026-06-02 19:27:07 +00:00

ops

Fix CuTeDSL from_dlpack device mismatch inside CUDA graph capture

2026-06-03 20:34:24 +00:00

reference

Cleanup Step 2: Archive Lineage P code, fix broken imports

2026-06-02 19:27:07 +00:00

__init__.py

Restructure: cutedsl/ -> dsv4/ with proper layering

2026-05-21 17:30:44 +00:00