How to re-decompose operator in Inductor pass

I want to add a pass re-decompose operators in Inductor, I am wondering how to do it:

  • set “decomposition=None” in compile_fx(), the forward graph and backward graph traced are not decomposed.
  • add a pass in post_grad_passes(), which can re-decompose the operators using decomposition_table.