From faab7536fc62096b7fd16501be6d39f996fabf12 Mon Sep 17 00:00:00 2001 From: Yujia Zhai Date: Sat, 17 Sep 2022 18:40:30 -0700 Subject: [PATCH] add comment (#628) --- examples/41_multi_head_attention/fused_multihead_attention.cu | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/41_multi_head_attention/fused_multihead_attention.cu b/examples/41_multi_head_attention/fused_multihead_attention.cu index 455e6284..ae3b66d3 100644 --- a/examples/41_multi_head_attention/fused_multihead_attention.cu +++ b/examples/41_multi_head_attention/fused_multihead_attention.cu @@ -43,6 +43,8 @@ # Run an attention example with batch size = 64 and head number = 16 without checking the correctness $ ./examples/41_multi_head_attention/41_multi_head_attention --head_number=16 --batch_size=64 --reference-check=false + Acknowledgement: this example is inspired by the idea originally prototyped by ByteDance Inc. + */ /////////////////////////////////////////////////////////////////////////////////////////////////