passing subgroup_size to get_mem_access_map (now required)