Fix kernel arg setting and workgroup size