triton is all you need 之 GEMM

代码参考了傅哥,请b站关注我是傅傅猪喵,谢谢喵! Triton DSL是以BLOCK tile为中心的Python DSL。与CUDA相比,Triton的使用者无法控制所有细节,因为某些优化是自动完成的,但是在Triton编译器的逐层编译优化之下也可以获得与Cuda相近甚至超过的性能。另外,Triton的编写和调试更加简单,而且学习成本更低。 ...

September 7, 2025 · 5 min

【CUDA从入门到入土】四、矩阵乘法

矩阵乘法跟之前不同,之前一维可以直接写一个kernel,或者多个kernel线性的排布来并行计算,那么矩阵乘法就是由一维向二维转变的关键。这时候一维的kernel排布也变成了二维排布。 ...

September 3, 2025 · 1 min

【CUDA从入门到入土】三、reduce算子及其优化

该项目代码参考傅哥的课程,很有用的课程,请多多支持他。 reduce 规约求和是cuda中一个经典的问题,其本质是将输入的序列进行求和。 在CUDA的多线程中,我们清楚数据被分为一个一个的block中进行运行,每个block通过warp来并发32个线程进行运算。 ...

August 23, 2025 · 8 min

【CUDA从入门到入土】二、CUDA调试和必知必会 & Nsight Computer 入门

上文中,我们运行了一个简单的cuda函数,并且一次过的将其运行了起来,这次,我们需要补充一些基础的概念,通过概念和框架的建立,我们才能走的更远,高屋建瓴的认识更多。 ...

August 17, 2025 · 7 min

【CUDA从入门到入土】五、cuda_kernel_和_cuda_attention详解

从cuda kernel出发,看懂人生第一个cuda attention

August 17, 2025 · 1 min

【CUDA从入门到入土】一、丝滑的CUDA入门

CUDA是什么 cuda是一种gpu编程组件,是一种原生支持GPU软硬件的架构,使得开发者可以直接在 GPU 上编写和执行通用计算程序。 GPU架构 上图是H100白皮书中,H100 GPU带满了144个SM的架构图 上图是H100中,1个SM的架构图 SM架构详解由此,我们可以高屋建瓴的看懂GPU,这里不雕琢细节,只直白的说明白GPU怎么工作的: PCIE 就是负责数据从显卡和主板之间进出的数据传输协议,理解为传递数据的就对了 ...

August 14, 2025 · 8 min

WSL2搭建cuda-triton开发环境

先用了vmware + ubuntu的方法,结果发现现在gpu无法透传到vmware的虚拟机里 故此使用wsl2的开发环境,后续会更新许多更舒服丝滑的操作 WSL2查看所有发行版本 wsl --list --online 安装指定版本 wsl --install Ubuntu-22.04 安装成功之后设置一下账户密码就行 安装miniconda用脚本安装 wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh chmod +x Miniconda3-latest-Linux-x86_64.sh ./Miniconda3-latest-Linux-x86_64.sh source ~/.bashrc conda list 用脚本安装就一直yes就行 安装CUDA Toolkit 和 cudnn使用 nvidia-smi查看cuda版本,在下面链接选择相应的版本 ...

August 14, 2025 · 4 min