A Code Generator for High-Performance Tensor Contractions on GPUs.

Jinsung Kim, Aravind Sukumaran-Rajam, Vineeth Thumma, Sriram Krishnamoorthy, Ajay Panyala, Louis-Nol Pouchet, Atanas Rountev, P. Sadayappan

Browse the full CGO paper archive.