Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.

An extension library of WMMA API (Tensor Core API)

FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme

Artifact for PPoPP22 QGTC: Accelerating Quantized GNN via GPU Tensor Core.

Fast SGEMM emulation on Tensor Cores

Fast Kernel SVM on TensorCore enabled GPU

An extension library of WMMA API for single precision matrix operation using TensorCores and error correction technique

A software TensorCore using warp shuffle

Compare the different runtime of CNN computation on CPU and GPU

Artifact for SC21: APNN-TC: Accelerating Arbitrary Precision Neural Networks on Ampere GPU Tensor Cores.

Experiments to accelerate GPU device for PyTorch training

simple examples of tools and libraries