Spaces:
Sleeping
Sleeping
/****************************************************************************** | |
* Copyright (c) 2023, Tri Dao. | |
******************************************************************************/ | |
// Split into multiple files to compile in paralell | |
#include "selective_scan_fwd_kernel.cuh" | |
template void selective_scan_fwd_cuda<at::Half, float>(SSMParamsBase ¶ms, cudaStream_t stream); | |
template void selective_scan_fwd_cuda<at::Half, complex_t>(SSMParamsBase ¶ms, cudaStream_t stream); |