xma.functional.softmax.triton_implementation.backward

softmax_backward_triton_kernel(y_ptr, y_stride, dy_ptr, dy_stride, dx_ptr, dx_stride, logits_multiplier, B, H, BLOCK_SIZE_B: triton.language.constexpr, BLOCK_SIZE_H: triton.language.constexpr)