https://github.com/state-spaces/mamba/blob/34076d664838588a3c97727b263478ab9f621a07/csrc/selective_scan/selective_scan_fwd_kernel.cuh#L82 pay attention to anything with smem or shared.