YuxueYang1204 / CudaDemo
Implement custom operators in PyTorch with cuda/c++
☆59Updated 2 years ago
Alternatives and similar repositories for CudaDemo:
Users that are interested in CudaDemo are comparing it to the libraries listed below
- Examples of CUDA implementations by Cutlass CuTe☆170Updated 3 months ago
- A tutorial for CUDA&PyTorch☆138Updated 3 months ago
- learning how CUDA works☆247Updated 2 months ago
- A light llama-like llm inference framework based on the triton kernel.☆112Updated this week
- Codes & examples for "CUDA - From Correctness to Performance"☆98Updated 6 months ago
- llm theoretical performance analysis tools and support params, flops, memory and latency analysis.☆87Updated 3 months ago
- A minimalist and extensible PyTorch extension for implementing custom backend operators in PyTorch.☆33Updated last year
- 📚FFPA(Split-D): Yet another Faster Flash Attention with O(1) GPU SRAM complexity large headdim, 1.8x~3x↑🎉 faster than SDPA EA.☆171Updated last month
- ☆139Updated 4 months ago
- Tutorials for writing high-performance GPU operators in AI frameworks.☆130Updated last year
- A simple high performance CUDA GEMM implementation.☆366Updated last year
- hands on model tuning with TVM and profile it on a Mac M1, x86 CPU, and GTX-1080 GPU.☆47Updated last year
- ☆123Updated last year
- 大规模并行处理器编程实战 第二版答案☆32Updated 2 years ago
- Personal Notes for Learning HPC & Parallel Computation [Active Adding New Content]☆66Updated 2 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆346Updated 7 months ago
- 分层解耦的深度学习推理引擎☆72Updated 2 months ago
- CUDA 算子手撕与面试指南☆326Updated 3 months ago
- ☆22Updated last month
- flash attention tutorial written in python, triton, cuda, cutlass☆341Updated 4 months ago
- Implement Flash Attention using Cute.☆78Updated 4 months ago
- Optimize softmax in triton in many cases☆20Updated 8 months ago
- ☆117Updated 5 months ago
- 先进编译实验室的个人主页☆81Updated 2 weeks ago
- ☆20Updated last month
- Machine Learning Compiler Road Map☆44Updated last year
- 使用 CUDA C++ 实现的 llama 模型推理框架☆54Updated 5 months ago
- 《CUDA编程基础与实践》一书的代码☆120Updated 3 years ago
- Triton Documentation in Chinese Simplified / Triton 中文文档☆67Updated 3 weeks ago
- Parallel Prefix Sum (Scan) with CUDA☆19Updated 10 months ago