Open jialiangZ opened 2 weeks ago
out, x, *rest = selective_scan_cuda.fwd(u, delta, A, B, C, D, z, delta_bias, delta_softplus) I'm wondering if this final_state refers to the state depicted by the red circle in the figure
out, x, *rest = selective_scan_cuda.fwd(u, delta, A, B, C, D, z, delta_bias, delta_softplus) I'm wondering if this final_state refers to the state depicted by the red circle in the figure