How to implement lookup table in codegen?

My acceleration chip can calculate the activation function through fast lookup table.I know how to call hardware instructions through tensorzise scheduling primitives, but I don’t know how to describe the calculation process on the front end (topi). This prevents me from assigning tables from host to device. Thanks in advance for any help.