tropicalgpu

TropicalGEMM on GPU

Difficulty level

Advanced

Advisor

Jinguo Liu ([email protected])

Project Description

Using CUDA.jl to implement matrix multiplication (GEMM) for tropical numbers. In mathematics, tropical algebra is a semiring with the addition replaced with maximization and the multiplication replaced with ordinary addition:

$$\begin{align} &x \oplus y = \max\{x, y\},\\\ &x \otimes y = x + y. \end{align}$$

One can find an implementation of tropical GEMM for CPU here: https://github.com/TensorBFS/TropicalGEMM.jl . However, a CUDA implementation is still missing.

References

• Using case: Tropical Tensor Network for Ground States of Spin Glasses
• Libraries: CUDA.jl and TropicalNumbers.jl

Project Output Standard

A working Julia implementation for Tropical GEMM on GPU, with better performance than the existing naive implementation.

Technical requirements

• CUDA programming (professional),
• Julia language (basic).