espnet2.asr_transducer.decoder.modules.rwkv.attention.load_wkv_kernel

source

espnet2.asr_transducer.decoder.modules.rwkv.attention.load_wkv_kernel(context_size: int) → None

Load WKV CUDA kernel.

• Parameters:context_size – Context size.