TensorRT-LLMs/cpp/tensorrt_llm/kernels/contextFusedMultiHeadAttention
Kaiyu Xie 75b6210ff4
Kaiyu/update main (#5)
* Update

* Update
2023-10-18 22:38:53 +08:00
..
cubin Kaiyu/update main (#5) 2023-10-18 22:38:53 +08:00
fmhaRunner.cpp Kaiyu/update main (#5) 2023-10-18 22:38:53 +08:00
fmhaRunner.h Kaiyu/update main (#5) 2023-10-18 22:38:53 +08:00
fused_multihead_attention_common.h Kaiyu/update main (#5) 2023-10-18 22:38:53 +08:00
fused_multihead_attention_v2.h Initial commit 2023-09-20 00:29:41 -07:00
tmaDescriptor.h Initial commit 2023-09-20 00:29:41 -07:00