You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
# Create offset tile
offsets = ct.add(ct.arange(BLOCK_SIZE, dtype=ct.int32), block_start) # new var
mask = ct.less(offsets, n_elements) # new var
# Load input data
x_tile = ct.gather(x, offsets) # new var
TileGym/src/tilegym/ops/cutile/activation/gelu.py
Line 125 in 98175fd