xma.functional.swiglu.cuda_implementation.backward

swiglu_backward_cuda_jit(mG: cutlass.cute.Tensor, mU: cutlass.cute.Tensor, mdY: cutlass.cute.Tensor, mdG: cutlass.cute.Tensor, mdU: cutlass.cute.Tensor) → None

swiglu_backward_cuda_kernel(gG: cutlass.cute.Tensor, gU: cutlass.cute.Tensor, gdY: cutlass.cute.Tensor, gdG: cutlass.cute.Tensor, gdU: cutlass.cute.Tensor, gID: cutlass.cute.Tensor, copy_atom: cutlass.cute.CopyAtom, tiled_copy: cutlass.cute.TiledCopy, shape: cutlass.cute.Shape) → None