colossalai.kernel.jit

colossalai.kernel.jit.set_jit_fusion_options()

Set PyTorch JIT layer fusion options.

• colossalai.kernel.jit.bias_dropout_add
• colossalai.kernel.jit.bias_gelu
• colossalai.kernel.jit.option