Fix test cases that require work-group size > 1