1. fix error because of unsupported node type. 2. fix check logic to get rank id in distribution training case.