CUB requires static shapes, which means TIDx must be static. It's likely to make more sense to pad to warp boundaries.