ME/CS 759 Final Project

Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction

Links

• ME/CS 759 Main repo: https://github.com/ShawnZhong/CS759-Spring-2020

Folder Structure

• Reports

  • Final Project.pdf

  • Intermediate Report.pdf

  • Project Proposal.pdf

• Code

  • benchmark: benchmark scripts and results

  • cutensor: cuTENSOR implementation of einsum

  • path_opt: path searching algorithm, implemented by RobeZH

  • specialization: einsum specialization, implemented by YuhanLiu11

Other CS/ME 759 Final Projects Available on GitHub

• Spring 2020

  • https://github.com/dgschmitz17/ME759_FinalProject_BlankSchmitz

  • https://github.com/elisaoh/ECE759

  • https://github.com/Kaiweitu/ME759_Final_Project

  • https://github.com/NoahMattV/image-upscaling

  • https://github.com/realXGong/ME759-final

  • https://github.com/sandersyen/ME759-Final-Project

  • https://github.com/schen921/me759finalproject

  • https://github.com/tlfua/ME759_final_project

  • https://github.com/wu0607/2020-Spring-ME759-FinalProject

• Past projects

  • https://github.com/BinQuake/CS759_final_proj

  • https://github.com/collielimabean/ME759-F17-FinalProject

  • https://github.com/devinsm/cs759Final

  • https://github.com/elliottbiondo/ME759_final_project

  • https://github.com/JerryHu1994/GPU-Implementation-of-Order-Independent-Pixel-based-Texture-Synthesis

  • https://github.com/Kagami-sama/UWMadison_ME759_FinalProject

  • https://github.com/krovchen/ME759_FinalProject

  • https://github.com/ljyang17/759finalproject

  • https://github.com/paragpathak2006/ME759FinalProject

  • https://github.com/raunakbardia/GALS_Advection_ME759

  • https://github.com/sapangupta/759-sapangupta-finalproject

  • https://github.com/tvdmitrii/ME759_FinalProject

  • https://github.com/WenFuLee/CS-759-High-Performance-Computing