首页 发现 帮助
注册 登录
david
/
aphrodite-engine
镜像自地址 https://github.com/PygmalionAI/aphrodite-engine
1
0
派生 0
文件 工单管理 0 Wiki
目录树: e177788401
分支列表 标签列表
aphrodite-engin...
/
kernels
/
punica
/
bgmv
/
bgmv_bf16_bf16_bf16.cu

bgmv_bf16_bf16_bf16.cu 130 B
文件历史 原始文件

1234 
  1. #include "bgmv_config.h"
    2. 

  2. #include "bgmv_impl.cuh"
    3. 

  3. 

  4. FOR_BGMV_WIDE_NARROW(INST_BGMV_TWOSIDE, nv_bfloat16, nv_bfloat16, nv_bfloat16)
    5.