[D] I wrote a small tool for debugging Triton code. Anyone interested?

Hey, I am writing Triton kernels and the only way to debug the code as far as I know, is using tl.device\_print, which only works with tensor data (no shapes for you) and clogs the output. So I wrote a small tool to run kernels using just torch without changing the code. The only changes is reducing launch grid sizes and changing kernel wrappers to debug wrappers. Here's an example of a simple kernel:

import torch
import triton
# import …

code data debug debugging hey machinelearning small tensor tool torch triton writing