Folks might find the author’s research paper [1] while at Harvard more informative. This is a great high-level description, but if you want more detail, I recommend the paper. [1] https://dl.acm.org/doi/abs/10.1145/3315508.3329973
That's http://www.eecs.harvard.edu/~htk/publication/2019-mapl-tille... for those of us outside the paywall
Triton: Open-Source GPU Programming for Neural Networks
81–90 of 116 posts
Re: Triton: Open-Source GPU Programming for Neural Networks
#82So the code looks (apart from pointers) similiar to numba which feels much closer to numpy/pytorch high level code. Are there huge advantages in the triton model compared to numba that I don't see? Or is there a big performance gap? For me numba was always the easiest way to get some new idea running on cuda, and most of the time it was fast enough.. Did anybody find performance comparison between numba and triton?
I guess Triton is not for you if Numba is fast enough.
Re: Triton: Open-Source GPU Programming for Neural Networks
#83Earlier quoted context omitted.
I believe this is more of an optimization layer to be utilized by libraries like Tensorflow and JAX. More of a simplification of the interaction with traditional CUDA instructions. I imagine these libraries and possibly some users would implement libraries on top of this language and reap some of the optimization benefit without having to maintain low-level CUDA specific code.
Im curious how it would compare to Halide Lang. They both seem to be targetting the same problem.
Re: Triton: Open-Source GPU Programming for Neural Networks
#84Earlier quoted context omitted.
AMD's ROCm 4.0 now supports cooperative groups, which is probably one of the last major holdouts for CUDA compatibility. There's still the 64-wavefront (for AMD CDNA cards) instead of 32-wavefronts (for CUDA). But AMD even has 4x4 half-float matrix multiplication instructions in ROCm (for MI100, the only card that supports the matrix-multiplication / tensor instructions) --------- I think CUDA vs OpenCL is over. ROCm…
Two reasons also. Some of the fastest research supercomputers in the US have been nVIDIA based. Sitting at a national lab, it's laughable how much more support there is from nVIDIA. Getting a DGX box is easy - there is nothing comparable from AMD's side. And the less said about Intel it's better. Smart, motivated people I know have left Argonne recently because of the Aurora shitshow.
Re: Triton: Open-Source GPU Programming for Neural Networks
#85Re: Triton: Open-Source GPU Programming for Neural Networks
#86Earlier quoted context omitted.
The author commented on reddit about that ( https://www.reddit.com/r/MachineLearning/comments/otdpkx/n_i... ) > PS: The name Triton was coined in mid-2019 when I released my PhD paper on the subject ( http://www.eecs.harvard.edu/~htk/publication/2019-mapl-tille... ). I chose not to rename the project when the Triton inference server came out a year later since it's the only thing that ties my helpful PhD advisors to…
The author is unfortunately wrong. NVIDIA's Triton was referenced in marketing material as far back as 2018. https://developer.nvidia.com/blog/nvidia-serves-deep-learnin...
Same industry: check
Maybe the author should expect an incoming C&D. IIRC US is first-to-use, so NVIDIA would prevail?
Re: Triton: Open-Source GPU Programming for Neural Networks
#87Earlier quoted context omitted.
Two reasons also. Some of the fastest research supercomputers in the US have been nVIDIA based. Sitting at a national lab, it's laughable how much more support there is from nVIDIA. Getting a DGX box is easy - there is nothing comparable from AMD's side. And the less said about Intel it's better. Smart, motivated people I know have left Argonne recently because of the Aurora shitshow.
E.g. Pavan lol
Re: Triton: Open-Source GPU Programming for Neural Networks
#88Earlier quoted context omitted.
Two reasons also. Some of the fastest research supercomputers in the US have been nVIDIA based. Sitting at a national lab, it's laughable how much more support there is from nVIDIA. Getting a DGX box is easy - there is nothing comparable from AMD's side. And the less said about Intel it's better. Smart, motivated people I know have left Argonne recently because of the Aurora shitshow.
Can you comment more on what’s going on at Argonne related to Aurora?
Intel, last I heard has written off close to 300 million dollars on Aurora.
Re: Triton: Open-Source GPU Programming for Neural Networks
#89Unfortunate name clash with NVIDIAs Triton Inference Server: https://developer.nvidia.com/nvidia-triton-inference-server
Re: Triton: Open-Source GPU Programming for Neural Networks
#90This sounds like basically hand-holding for Python programmers to write simple NN-operations. I'm sure it's convenient and useful, but it's still glorified glue code.