xla

Fix a corner case where view tensor update will be ignore by mark_step

#4232

Merged