swr/rast: Move memory-related JIT functions