MCPcopy
hub / github.com/state-spaces/mamba / backward

Method backward

mamba_ssm/ops/triton/k_activations.py:166–168  ·  view source on GitHub ↗
(ctx, dout)

Source from the content-addressed store, hash-verified

164
165 @staticmethod
166 def backward(ctx, dout):
167 xy, = ctx.saved_tensors
168 return _swiglu_bwd(xy, dout)
169
170
171swiglu = SwiGLU.apply

Callers 9

_runFunction · 0.45
test_selective_scanFunction · 0.45
test_mamba_inner_fnFunction · 0.45
test_mamba3_siso_combined_batchedFunction · 0.45
test_mamba3_siso_combined_varlenFunction · 0.45
test_layer_norm_gatedFunction · 0.45
test_mamba_mimo_smoke_forward_backwardFunction · 0.45
test_mamba_mimo_smoke_forward_backward_varlenFunction · 0.45
test_mamba3_varlen_backwardFunction · 0.45

Calls 1

_swiglu_bwdFunction · 0.85

Tested by 9

_runFunction · 0.36
test_selective_scanFunction · 0.36
test_mamba_inner_fnFunction · 0.36
test_mamba3_siso_combined_batchedFunction · 0.36
test_mamba3_siso_combined_varlenFunction · 0.36
test_layer_norm_gatedFunction · 0.36
test_mamba_mimo_smoke_forward_backwardFunction · 0.36
test_mamba_mimo_smoke_forward_backward_varlenFunction · 0.36
test_mamba3_varlen_backwardFunction · 0.36