modelopt.torch.quantization.backends.gemm_registry
- gemm_registry = <modelopt.torch.quantization.backends.gemm_registry.GEMMRegistry object>
Registry for specialized GEMM (General Matrix Multiplication) implementations.
This registry allows registering specialized GEMM implementations with custom availability checks, giving users control over matching criteria.