ardenma / implicit-gemm-tensor-core-convolution

Simple example of how to write an Implicit GEMM Convolution in CUDA using the tensor core WMMA API and bindings for PyTorch.
18Updated last year

Alternatives and similar repositories for implicit-gemm-tensor-core-convolution:

Users that are interested in implicit-gemm-tensor-core-convolution are comparing it to the libraries listed below