fix upcast in slice attention (#1591)

Commit

3 years ago

fix upcast in slice attention (#1591) * fix upcast in slice attention * fix dtype * add test * fix test