DeepSpeed
Add conditional on torch version for scaled_dot_product_attention
#6517
Merged
Merged
Add conditional on torch version for scaled_dot_product_attention #6517
loadams
merged
3 commits into
master
from
loadams/flops-profiler-scaled-dor-attn-torch-2
Add conditional on torch version for scaled_dot_product_attention
d5ad09d0
Merge branch 'master' into loadams/flops-profiler-scaled-dor-attn-tor…
039cb2f9
Merge branch 'master' into loadams/flops-profiler-scaled-dor-attn-tor…
873d0b1a
tohtana
approved these changes
on 2024-09-11
Login to write a write a comment.
Login via GitHub