When the second argument contains batch dimension, the axis should be 0.

Also makes elementwise ops more tolerant at handling tensors with trailing
singular dimensions.

* merge fluid dist tests

* update cmake

* Fix dist demo error

* revert trainer_id

* Enhance bipartite_match_op to support argmax matching after bipartite matching.

* Fix typo error.