add comment (#628)

2026-05-12 01:10:08 +00:00 · 2022-09-17 18:40:30 -07:00
parent fc9ebc645b
commit faab7536fc
1 changed files with 2 additions and 0 deletions
--- 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.
+
 */

 /////////////////////////////////////////////////////////////////////////////////////////////////