Add conditional on torch version for scaled_dot_product_attention - #6517

Add conditional on torch version for scaled_dot_product_attention#6517

loadams merged 3 commits into

masterdeepspeedai/DeepSpeed:masterfrom

loadams/flops-profiler-scaled-dor-attn-torch-2deepspeedai/DeepSpeed:loadams/flops-profiler-scaled-dor-attn-torch-2Copy head branch name to clipboard