pytorch
918054f4 - [Inductor] support channel last for xpu conv in inductor layout opt path (#111018)

Commit
1 year ago
[Inductor] support channel last for xpu conv in inductor layout opt path (#111018) # Motivation support xpu channel last for inductor layout optimization path. Currently, `_conv_determine_backend_memory_format` always returns torch.contiguous_format for XPU conv. # Solution Add xpu channel last detection stragey in `determine_backend_memory_format` Pull Request resolved: https://github.com/pytorch/pytorch/pull/111018 Approved by: https://github.com/jansel, https://github.com/eellison, https://github.com/EikanWang
Author
Committer
Parents
Loading