xla icon indicating copy to clipboard operation
xla copied to clipboard
verified publisher icon pytorch

Add IsVirtualDeivce() and refactor

Open yeounoh opened this issue 1 year ago • 0 comments

This addresses the following to support #6322

  • add IsVirtualDeivce()
  • Un-const tensors argument to XLAGraphExecutor::Compile method, since they can be updated during auto-sharding paass.
  • refactor to remove redundant codes

yeounoh avatar Mar 12 '24 20:03 yeounoh