xla
e2389eaf - [Functionalization] Lower as_strided_scatter properly (#4770)

Commit
2 years ago
[Functionalization] Lower as_strided_scatter properly (#4770) Summary: The current lowering of as_strided_scatter is using a hack that utilize our old view infra to reapply value updates to the base view. Let's lower it properly by using AsStridedViewUpdate. Test Plan: PJRT_DEVICE=CPU python ../test/test_view_ops.py -v -k TestViewOpsXLA.test_as_strided
Author
Parents
Loading