add grouped train v2

b4a995b8 · danyao12 · d8d8c6f7 · b4a995b8
Commit b4a995b8 authored Jun 16, 2023 by danyao12
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 1031 additions and 0 deletions

example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_train_v2.cpp ...ale_softmax_gemm/grouped_multihead_attention_train_v2.cpp +1031 -0

No files found.
--- a/example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_train_v2.cpp
+++ b/example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_train_v2.cpp