NVIDIA / HMM_sample_code
CUDA 12.2 HMM demos
☆19Updated 9 months ago
Alternatives and similar repositories for HMM_sample_code:
Users that are interested in HMM_sample_code are comparing it to the libraries listed below
- 方便扩展的Cuda算子理解和优化框 架,仅用在学习使用☆13Updated 10 months ago
- A memory profiler for NVIDIA GPUs to explore memory inefficiencies in GPU-accelerated applications.☆25Updated 6 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆82Updated this week
- GPTQ inference TVM kernel☆38Updated last year
- Standalone Flash Attention v2 kernel without libtorch dependency☆108Updated 7 months ago
- FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme☆59Updated last month
- TiledKernel is a code generation library based on macro kernels and memory hierarchy graph data structure.☆19Updated 11 months ago
- Benchmark tests supporting the TiledCUDA library.☆16Updated 5 months ago
- Code for Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB).The outdated wr…☆9Updated last year
- An Attention Superoptimizer☆21Updated 3 months ago
- Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt, but ported to both NVIDIA and AMD GPUs.☆32Updated 3 weeks ago
- ☆50Updated last year
- An extension library of WMMA API (Tensor Core API)☆96Updated 9 months ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆91Updated 6 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆91Updated 3 weeks ago
- An IR for efficiently simulating distributed ML computation.☆28Updated last year
- RCCL Performance Benchmark Tests☆64Updated this week
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆50Updated 9 months ago
- ☆78Updated 5 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆73Updated 3 weeks ago
- ☆55Updated 2 weeks ago
- Framework to reduce autotune overhead to zero for well known deployments.☆65Updated last week
- rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.☆77Updated this week
- A practical way of learning Swizzle☆18Updated 2 months ago
- ☆21Updated last year
- ☆9Updated last year
- ☆26Updated last year
- Ahead of Time (AOT) Triton Math Library☆57Updated last week
- An external memory allocator example for PyTorch.☆14Updated 3 years ago
- ☆95Updated last year