TrueHero

icon company linkedin icon location Sunnyvale, CA

Results 1 issues of TrueHero

Where to see the CUDA source code of these CUDA kernels?

2 comment

Hi team, I am interested to see the CUDA source code of these transformer kernels, like: https://github.com/NVIDIA/TensorRT/blob/e5f9ead4a4826cc774325720a26dbf4ec47203ea/plugin/bertQKVToContextPlugin/fused_multihead_attention/src/fused_multihead_attention_fp16_128_64_kernel.sm75.cpp Anyone know where to see the source code?

Samples
triaged