Home Esplora Aiuto
Registrati Accedi
david
/
aphrodite-engine
mirror da https://github.com/PygmalionAI/aphrodite-engine
1
0
Forka 0
File Problemi 0 Wiki
Albero (Tree): afb710e3e4
Rami (Branch) Tag
aphrodite-engin...
/
kernels
/
punica
/
bgmv
/
bgmv_fp32_fp16_bf16.cu

bgmv_fp32_fp16_bf16.cu 194 B
Cronologia Originale

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

  2. #include "bgmv_impl.cuh"
    3. 

  3. 

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

  5. FOR_INST_BGMV_WIDE_NARROW(INST_BGMV_ONESIDE, float, nv_half, nv_bfloat16)
    6.