Fix workgroup size. AMD gpus can't run workgroups larger than 256.
Fix workgroup size. AMD gpus can't run workgroups larger than 256.

lift-project/lift,lift-project/lift,lift-project/lift,lift-project/lift,lift-project/lift