make dy contiguous

2026-01-14 11:10:59 +00:00 · 2026-01-14 11:10:59 +00:00 · 894828cfb8
commit 894828cfb8
parent 1d86812b7c
1 changed files with 1 additions and 0 deletions
--- a/unsloth/kernels/moe/grouped_gemm/interface.py
+++ b/unsloth/kernels/moe/grouped_gemm/interface.py
@ -746,6 +746,7 @@ class GroupedGemm(torch.autograd.Function):

    @staticmethod
    def backward(ctx, dY):
+        dY = dY.contiguous()
        X, W, m_sizes, gather_indices = ctx.saved_tensors
        topk = ctx.topk
        permute_x = ctx.permute_x