gpu implemention of kmeans