cutlass
triton
Our great sponsors
cutlass | triton | |
---|---|---|
16 | 30 | |
4,522 | 10,981 | |
6.1% | 7.1% | |
8.8 | 9.9 | |
3 days ago | about 5 hours ago | |
C++ | C++ | |
GNU General Public License v3.0 or later | MIT License |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
cutlass
-
Optimization Techniques for GPU Programming [pdf]
I would recommend the course from Oxford (https://people.maths.ox.ac.uk/gilesm/cuda/). Also explore the tutorial section of cutlass (https://github.com/NVIDIA/cutlass/blob/main/media/docs/cute/...) if you want to learn more about high performance gemm.
-
Want to understand INT8 better
The latter (and I guess you were asking about this one) is designed to accelerate NN inference in reduced precision. It is possible to use Tensor Cores for you own purposes, mainly through CUTLASS. But because Tensor Cores are designed to execute matrix multiplications, it can be hard to adapt your problem to them. The performance with them is insane (IIRC 32x the performance of the INT32 pipeline), but only for matrix multiplication…
-
How do I deal with tensor core and cuda core with different precision?
If you want to learn about controlling Tensor Cores, the main way is through the CUTLASS library, that wraps the complexity of Tensor Cores into higher level abstractions. You can also look for mma/wmma instructions in the PTX specification, or for the WMMA API in CUDA.
-
AI’s compute fragmentation: what matrix multiplication teaches us
> we used tensor cores and managed to get back fp32 accuracy with 3 rounds of the things
Hey are you referring to 3xTF32 (https://github.com/NVIDIA/cutlass/tree/master/examples/28_am...)? IMO this is a perfect example where proper abstraction could save engineers non-trivial amount of time - imagine a compiler stack which allows 3xTF32 as a normal dtype and subsequent analysis compatible with this special dtype :-)
- With LLVM and MLIR, is manual cuda optimizing still important?
- CUTLASS 3.0 is now available
-
How to Optimize a CUDA Matmul Kernel for CuBLAS-Like Performance: A Worklog
This is a great post for people who are new to optimizing GPU code.
It is interesting to see that the author got this far without interchanging the innermost loop over k to the outermost loop, as is done in CUTLASS (https://github.com/NVIDIA/cutlass).
As you can see in this blog post the code ends up with a lot of compile-time constants (e.g. BLOCKSIZE, BM, BN, BK, TM, TN) one way to optimize this code further is to use an auto-tuner to find the optimal value for all of these parameters for your GPU and problem size, for example Kernel Tuner (https://github.com/KernelTuner/kernel_tuner)
-
pytorch example to actually see anything near 83 TFLOP/s on a RTX 4090?
Some examples here have a benchmark: https://github.com/NVIDIA/cutlass/blob/master/examples/24_gemm_grouped/gemm_grouped.cu
-
Create a bare CMake for Nvidia CUTLASS
I would like to make a minimum CMakeLists to use the CUDA CUTLASS library in another project. The build system is CMake, however I have little experience with CMake.
-
[D] What are some good resources to learn CUDA programming?
If you already know some C++, the Nvidia devblog is a great resource. Going further, Cub and Cutlass provide examples of efficient implementations for key operations at all hardware levels. Finally, this is more anecdotal but I always start my lectures on Cuda programming with the pictures in this doc page, to provide some intuition on the different memory layers that you can leverage to speed up a program. In any case, good luck :-)
triton
- OpenAI Triton: language and compiler for highly efficient Deep-Learning
-
Show HN: Ollama for Linux – Run LLMs on Linux with GPU Acceleration
There's a ton of cool opportunity in the runtime layer. I've been keeping my eye on the compiler-based approaches. From what I've gathered many of the larger "production" inference tools use compilers:
- https://github.com/openai/triton
- Core Functionality for AMD #1983
- Project name easily confused with Nvidia triton
-
Nvidia's CUDA Monopoly
Does anyone have more inside knowledge from OpenAI or AMD on AMDGPU support for Triton?
I see this:
https://github.com/openai/triton/issues/1073
But it's not clear to me if we will see AMD GPUs as first class citizens for pytorch in the future?
- @soumithchintala (Cofounded and lead @PyTorch at Meta) on Twitter: I'm fairly puzzled by $NVDA skyrocketing... (cont.)
-
The tiny corp raised $5.1M
I thought this was a good overview of the idea Triton can circumvent the CUDA moat: https://www.semianalysis.com/p/nvidiaopenaitritonpytorch
It also looks like they added MLIR backend to Triton though I wonder if Mojo has advantages since it was built on MLIR? https://github.com/openai/triton/pull/1004
-
Anyone hosting a local LLM server
I'm pretty happy with the setup, because it allows me to keep all the AI stuff and its dozens of conda envs and repos etc. seperate from my normal setup and "portable". It may have some performance impact (although I don't personally notice any significant difference to running it "natively" on windows), and it may enable some extra functionality, such as access to OpenAi's Triton etc., but that's currently neither here nor there.
- Triton: Runtime for highly efficient custom Deep-Learning primitives
-
Mojo – a new programming language for all AI developers
Very cool development. There is too much busy work going from development to test to production. This will help to unify everything. OpenAI Triton https://github.com/openai/triton/ is going for a similar goal. But this is a more fundamental approach.
What are some alternatives?
TensorRT - PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
cuda-python - CUDA Python Low-level Bindings
iree - A retargetable MLIR-based machine learning compiler and runtime toolkit.
Halide - a language for fast, portable data-parallel computation
GPU-Puzzles - Solve puzzles. Learn CUDA.
Chess_BinaryNeuralNetwork - Training and Code Emitting Library for Binary Neural Networks
dfdx - Deep learning in Rust, with shape checked tensors and neural networks
Open3D - Open3D: A Modern Library for 3D Data Processing
web-llm - Bringing large-language models and chat to web browsers. Everything runs inside the browser with no server support.
shark-samples
maxas - Assembler for NVIDIA Maxwell architecture