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
|