Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
activation.h 65 Bytes
Newer Older
Casper's avatar
MoE grouped gemm and fused topk_softmax (#8)  
Casper committed
1
2
3 
void silu_and_mul(
  torch::Tensor& out,
  torch::Tensor& input);