Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Kashif Rasul's avatar
    [MPS] call contiguous after permute (#1411) · babfb8a0
    Kashif Rasul authored 
    * call contiguous after permute

Fixes for MPS device

* Fix MPS UserWarning

* make style

* Revert "Fix MPS UserWarning"

This reverts commit b46c32810ee5fdc4c16a8e9224a826490b66cf49.
    babfb8a0
attention.py 37.5 KB