Fix treeAgg bug using ceil of log_2(partitions)
Also make normal equations match TSQR