Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction
- ME/CS 759 Main repo: https://github.com/ShawnZhong/CS759-Spring-2020
-
Reports
-
Code
-
benchmark: benchmark scripts and results
-
cutensor: cuTENSOR implementation of einsum
-
specialization: einsum specialization, implemented by YuhanLiu11
-