spikingjelly.activation_based.triton_kernel package

备注

Developers should decide whether Triton backend is available by:

try:
    import triton
except ImportError:
    triton = None

if triton is not None:
    # Triton backend is available
    ...
else:
    # Triton backend is not available
    ...

MemoryModule encapsulates this logic.

Predefined Neuron Kernels

• neuron_kernel
  • torch.compile Compatibility
  • IF
  • LIF
  • PLIF

FlexSN Implementation

• flexsn
  • Function Info
  • Template
  • Kernel Builders
  • Wrapper
  • Custom Ops
  • HigherOrderOp & Eager Scan

Torch-to-Triton Transpiler

• torch2triton
  • PyTorch to Graph
  • Graph to Triton

Spike Compressors

• compress

Utilities