Skip to content

Update benchmark_mha.py to compare with PyTorch SDPA#21449

Merged
tianleiwu merged 6 commits intomainfrom tlwu/benchmark_torch_mhaJul 27, 2024

Commits

Commits on Jul 23, 2024

Commits on Jul 26, 2024