Yunzhe Hao

icon company CASIA icon location Beijing China icon location PhD student @ CASIA; Main interests include Speech Separation, Auditory Attention, Spiking Neural Network

Results 2 issues of Yunzhe Hao

How to write cuda code of the multilayer units

1 comment

This tutorials helped me to write a single layer unit with CUDA code. But how to write CUDA code of the multilayer units, like torch/nn/_functions/rnn.py 281? ``` output, hy, cy,...

RuntimeError: FlashAttention only supports Ampere GPUs or newer.

0%| | 0/11 [00:00