bitorch_engine.layers.qlinear.binary.cutlass.layer

Classes

BinaryLinearCutlass(*args, **kwargs)	A specialized binary linear layer that leverages CUTLASS for efficient low-precision computations.
BinaryLinearForward(*args, **kwargs)	Implements the forward and backward passes of a binary linear layer.
BinaryMatMul([dtype])	A PyTorch module for binary matrix multiplication.
BinaryMatMulFunction(*args, **kwargs)	This class implements a custom autograd function for binary matrix multiplication.