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.cutlass
bitorch_engine.layers.qlinear.nbit.cutlass.extension
View page source
bitorch_engine.layers.qlinear.nbit.cutlass.extension
Functions
get_ext
Return CUDA extension for Q Linear Cutlass.