I’m trying to add support for new operator in Convert Layout. I’ve used this article and this pull request as a reference. I’m trying to implement Convert Layout pass for lightly-sensitive operator. For some reason I’m not getting new_in_layouts
when function is called, it’s not defined.
I’ve tested it by writing a test similar to test_transpose_convert_layout
, I just replaced transpose with the operator I’m trying to add. Am I supposed to add something more? Did something change since this article is written?