Home Explore Help
Register Sign In
david
/
aphrodite-engine
mirror of https://github.com/PygmalionAI/aphrodite-engine
1
0
Fork 0
Files Issues 0 Wiki
Branch: revert_871
Branches Tags
aphrodite-engin...
/
kernels
/
punica
/
bgmv
/
bgmv_bf16_bf16_bf16.cu

bgmv_bf16_bf16_bf16.cu 240 B
Permalink History Raw

123456 
  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. 

  5. FOR_INST_BGMV_WIDE_NARROW(INST_BGMV_ONESIDE, nv_bfloat16, nv_bfloat16,
    6. 

  6.                           nv_bfloat16)
    7.