Can we specify memory space for an input in TVM

IIUC, you’re looking for memory scopes to specify different memory types and cache_read and cache_write schedule primitives to move data between different memory types.

1 Like