Skip to content

vllm.model_executor.layers.fla.ops.op

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 
28
29
30
31
32
33
34
35
@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