Add explicit shmem id for data loading