Add missing code to enforce subgroup size in subgroup test.#2144