Skip to content

vllm.model_executor.layers.fla.ops.op

exp module-attribute

exp = fast_expf

log module-attribute

log = fast_logf

log2 module-attribute

log2 = fast_log2f

gather

gather(src, index, axis, _builder=None)

Gather operation that works when tl.gather is not supported. This is a fallback implementation that returns None. Just to make triton compiler happy.

Source code in vllm/model_executor/layers/fla/ops/op.py
@triton.jit
def gather(src, index, axis, _builder=None):
    """
    Gather operation that works when tl.gather is not supported.
    This is a fallback implementation that returns None.
    Just to make triton compiler happy.
    """
    return None

make_tensor_descriptor

make_tensor_descriptor(
    base, shape, strides, block_shape, _builder=None
)
Source code in vllm/model_executor/layers/fla/ops/op.py
@triton.jit
def make_tensor_descriptor(
    base,
    shape,
    strides,
    block_shape,
    _builder=None,
):
    return None