xla
Dynamo/AOTAutograd traceable flash attention
#8654

Merged

Dynamo/AOTAutograd traceable flash attention #8654

zpcore merged 4 commits into master from piz/autograde_trace

Support fw and bw with spmd

a8c8f470

zpcore force pushed from f82f3738 to a8c8f470 321 days ago

zpcore changed the title ~~backward with spmd issue~~ Dynamo/AOTAutograd traceable flash attention 321 days ago

nit

ff69d5a1

zpcore requested a review from

tengyifei 321 days ago

fix fw return signature

3bd4b1b0

tengyifei approved these changes on 2025-01-31

review update

de94ecb1

zpcore merged 9ae017e2 into master 320 days ago

zpcore deleted the piz/autograde_trace branch 320 days ago

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone