Using OpenGL instead of CUDA for machine learning

Specialist Fields Data Science Forum
, , , ,
#1

Summary

In this project, we

  1. Added an OpenGL backend for MXNet/TVM - a general-purpose tensor computation framework, so that it automatically compiles a Python program into an OpenGL shader that runs on the GPU on a computer that does not have CUDA.

  2. Explored optimizations of OpenGL shader programs so that a fundamental computation task needed in machine learning - matrix multiplication - has comparable performance with OpenCL on the same machine.

Read in full here:

https://phisiart.github.io/418-proj/

This thread was posted by one of our members via one of our news source trackers.

#2

Corresponding tweet for this thread:

Share link for this tweet.