mm: add context argument to shrinker callback