nv50/ir/nir: support load_work_dim