add gpu enabled KMeans