Use __local memory in NMS kernel and 16x16 workgroup size.