update KNN to be in parity with Trent's openCL changes.