[FEA] Calculating distances with FP32 in NN Descent

[jinsolp]

I hope NN Descent can support calculating distances in local_join_kernel using fp32 instead of fp16 (__half).

Due to precision issues, it is difficult to grab the distances calculated by NN Descent and compare it against fp32-calculated distances for additional operations outside of NN Descent such as making predictions.