kmeans_balanced.cuh and kmeans.cuh have some commonalities that can be abstracted away into helpers and reused. For example, the minClusterAndDistanceCompute function could likely be reused rather than calling the fused NN + Reduction from predict_core in kmeans_balanced.cuh
kmeans_balanced.cuhandkmeans.cuhhave some commonalities that can be abstracted away into helpers and reused. For example, theminClusterAndDistanceComputefunction could likely be reused rather than calling the fused NN + Reduction frompredict_coreinkmeans_balanced.cuh