mamba-ssm / selective-scan /selective_scan_bwd_fp32_complex.cu
danieldk's picture
danieldk HF Staff
Import mamba-ssm kernels
23d26f4
raw
history blame contribute delete
388 Bytes
/******************************************************************************
* Copyright (c) 2023, Tri Dao.
******************************************************************************/
// Split into multiple files to compile in paralell
#include "selective_scan_bwd_kernel.cuh"
template void selective_scan_bwd_cuda<float, complex_t>(SSMParamsBwd &params, cudaStream_t stream);