leslie-fang-intel 6616ad030f [Inductor] Fix the High Order Op layout issue (#128275) ... Fix the issue: https://github.com/pytorch/pytorch/issues/127995 - In current implementation of creating `FallbackKernel`, the `device` of the `NoneLayout` is set to `None` when `example_output` returns from `cls.process_kernel` is `None`. 921aa194c7/torch/_inductor/ir.py (L5632-L5649) - If a `ExternalKernel schedulerNode` has None device, the previous buffer will not flush before codegen this `ExternalKernel schedulerNode` which causes the wrong generated code. ef2b5ed500/torch/_inductor/scheduler.py (L2701-L2709) **Test Plan** ``` python -u -m pytest -s -v test/higher_order_ops/test_with_effects.py -k test_compile_inductor_external_op_return_none ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/128275 Approved by: https://github.com/eellison		2024-06-15 00:33:21 +00:00
..
test_with_effects.py	[Inductor] Fix the High Order Op layout issue (#128275)	2024-06-15 00:33:21 +00:00