Welcome to Triton-Viz, a visualization and profiling toolkit designed for deep learning applications. Built with the intention of making GPU programming on Triton more intuitive.
Table of Contents
Triton-Viz is a visualization and analysis toolkit specifically designed to complement the development and optimization of applications written in OpenAI's Triton, an open-source programming language aimed at simplifying the task of coding for accelerators such as GPUs. Triton-Viz offers a suite of features to enhance the debugging, performance analysis, and understanding of Triton code.
Given that Triton allows developers to program at a higher level while still targeting low-level accelerator devices, managing and optimizing resources like memory becomes a crucial aspect of development. Triton-Viz addresses these challenges by providing real-time visualization of tensor operations and their memory usage. The best part about this tool is that while it does focus on visualizing GPU operations, users are not required to have GPU resources to run examples on their system.
- Python installed (preferably the latest available version).
- Triton installed. Follow the installation instructions in the linked repository.
Upon successfully installing Triton, install Torch using the following command:
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121Upon successful installation of Torch make sure to uninstall pytorch-triton using the following command:
pip uninstall pytorch-tritonClone the repository to your local machine:
git clone https://github.com/Deep-Learning-Profiling-Tools/triton-viz.git
cd triton-viz
pip install -e .You're all set!
cd examples
python <file_name>.pyIf you're interested in fun puzzles to work with in Triton, do check out: Triton Puzzles
Triton-Viz is licensed under the MIT License. See the LICENSE for details.