DeepSpeed
2d84be81 - deepcompile: Create a full list of no-copy ops (#7562)

Commit
103 days ago
deepcompile: Create a full list of no-copy ops (#7562) The list of torch no-copy ops is hard coded and does not include all operations that may aliasing inputs in their outputs. Instead of using a fixed list, iterate over all ops under torch.ops.aten and identify those with aliasing behavior by inspecting their schema. With PyTorch 2.7.1, the default overload of ops identified by the updated logic include: - _nested_view_from_buffer - _reshape_alias - alias - as_strided - conj - detach - diagonal - expand - imag - lift_fresh - narrow - permute - pin_memory - positive - real - reshape - squeeze - t - unfold - unsqueeze - view - view_as_complex - view_as_real - most operations whose name ends with an underscore Signed-off-by: Junjie Mao <junjie.mao@linux.alibaba.com>
Author
Parents
Loading