⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, achieve peak⚡️ performance
-
Updated
Jan 13, 2025 - Cuda
#
tensor-cores
Here are 6 public repositories matching this topic...
📚[WIP] FFPA: Yet antother Faster Flash Prefill Attention with O(1)🎉GPU SRAM complexity for headdim > 256, 1.5x~2x🎉faster vs SDPA EA.
-
Updated
Jan 13, 2025 - Cuda
General Matrix Multiplication using NVIDIA Tensor Cores
matrix-multiplication cuda-kernels gpu-computing nvidia-cuda nvidia-gpu gpu-programming sgemm cuda-programming tensor-cores nvidia-tensor-cores
-
Updated
Oct 25, 2024 - Cuda
A benchmarking framework for correlators of FX telescope arrays
-
Updated
Oct 20, 2023 - Cuda
Vulkan & GLSL implementation of FlashAttention-2
vulkan glsl artificial-intelligence gpu-acceleration attention gpu-computing deel-learning tensor-cores large-language-models llm flash-attention flash-attention-2
-
Updated
Jan 12, 2025 - C++
Vulkan & GLSL implementation of FlashAttention-2
vulkan glsl artificial-intelligence gpu-acceleration attention gpu-computing deel-learning tensor-cores large-language-models llm flash-attention flash-attention-2
-
Updated
Jan 13, 2025
Improve this page
Add a description, image, and links to the tensor-cores topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the tensor-cores topic, visit your repo's landing page and select "manage topics."