srush / Triton-Puzzles

Puzzles for learning Triton
Apache License 2.0
925 stars 57 forks source link

ModuleNotFoundError: No module named 'triton_viz' #1

Closed neverix closed 5 months ago

neverix commented 5 months ago

The Colab tries to install the threed2 branch, which is not in the triton-viz repo.

srush commented 5 months ago

Haha, thanks for testing it out.

Should be fixed. That library is https://github.com/Deep-Learning-Profiling-Tools/triton-viz/ which is being developed concurrently with the puzzles.