unit_test on tree_prob, but it's much slower than cpu version. Add profiling, but it crashed when profiling inside forward_tree.