[FEA] cluster_cost can accept X norm

[tarang-jain]

The cluster_cost could be modified to accept an optional X_norm to avoid recomputing it. That function also synchronize each time it is called, while it would be better to accumulate cluster_cost on device and transfer+sync once at the end.