ardenma / implicit-gemm-tensor-core-convolutionLinks

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

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

Sorting: