Update klp_kmeans.py

GPU code set