Lu Teng

icon indicating an email [email protected]

icon company EVAS icon location China icon location AI Compiler Engineer at startup;　　　　　　　　　　　　 AI Framework Engineer @Intel, 2018~2024;　　　　　　　　　　　　　　　　　　　　　　　　　　　 Compiler Engineer @Huawei, 2012~2018.

Results 2 issues of Lu Teng

Add destructor for XLAGraphExecutor to free resource

1 comment

This PR is to add a destructor for `XLAGraphExecutor`, to trigger the destructor of PJRT class `Executable`/`LoadedExecutable`. This helps to do some special works (e.g. clear disk) when the PJRT...

Optimize as_strided_copy fast path to support offset

This PR optimizes the `as_strided_copy` fast path to support `offset`. Negative case (happens in the [default path of torch LSTM](https://github.com/pytorch/pytorch/blob/bc295ab6b48031712aeea16f48b994836952f0cf/aten/src/ATen/native/RNN.cpp#L752)): ```python import torch import torch_xla def torch_xla_chunk_example(input_tensor): chunks = torch.unsafe_chunk(input_tensor,...