Custom Gather-scatter Operator by CUTLASS
This blog is to log my experience of building efficient custom operator based on CUTLASS. Jump to the final implementation of gather and scatter matrix multi...
Recent Posts
Compact Inference with CUDA graph and StaticCache
This post is to log a minimum prototype of LLM inference with CUDA graph to eliminate bubbles between kernel launches. Click here to jump to the final implem...
Efficient Gather-and-scatter Matrix Multiplication Kernel with Triton
This post is to log my implementation of gather-and-scatter matrix multiplication operation with Triton. Click here to jump to the final implementation code.
Understand CUDA Unified Memory
This post is to log my experiments with CUDA unified memory and some innovative and interesting application of UVM in large language model (LLM).
Understand CUDA PTXAS
Placeholder for my blog