Bitorch Engine
Contents:
Installation
Build options
Full Documentation
bitorch_engine
bitorch_engine.functions
bitorch_engine.layers
bitorch_engine.layers.qconv
bitorch_engine.layers.qembedding
bitorch_engine.layers.qlinear
bitorch_engine.layers.qmha
bitorch_engine.optim
bitorch_engine.utils
Bitorch Engine
Full Documentation
bitorch_engine
bitorch_engine.layers
bitorch_engine.layers.qlinear
bitorch_engine.layers.qlinear.nbit
bitorch_engine.layers.qlinear.nbit.cuda
bitorch_engine.layers.qlinear.nbit.cuda.extension
View page source
bitorch_engine.layers.qlinear.nbit.cuda.extension
Functions
get_ext
Get the CUDA extension for quantized linear operations.