Skip to content
Back to pull request #504

Add unit tests for shared prefix masked attention with torch.FlexAttention #358

Sign in to view logs

Add unit tests for shared prefix masked attention with torch.FlexAttention

Add unit tests for shared prefix masked attention with torch.FlexAttention #358

Run time

Learn about OS pricing on GitHub Actions
Job Run time
checkstyle
7s
tests
11m 35s
tests-bwd
10m 12s
21m 54s