Add GPU support for our implementation of k-means.
Add GPU support for our implementation of k-means.