Skip to content

Refactor mkldnn-specific parts out of _inductor/overrides #90851

@ngimel

Description

@ngimel

Currently a lot of mkldnn-specific functions and classes reside in _inductor/overrides.py under fairly generic names such as ConvUnary2d

class ConvUnary2d(nn.Conv2d):
and fuse_unary
def fuse_unary(gm: torch.fx.GraphModule):
. This is confusing, and code should be structured to make it clear when mkldnn specific passes happen. Moving these functions and classes to a separate file clearly marked as mkldnn would be helpful.
cc @ezyang @soumith @msaroufim @wconstab @bdhirsh @mlazos @voznesenskym @yanboliang @penguinwu @anijain2305 @EikanWang @jgong5 @Guobing-Chen @chunyuan-w @XiaobingSuper @zhuhaozhe @blzheng @Xia-Weiwen @wenzhe-nrv @jiayisunx @peterbell10 @desertfire, @EikanWang

Metadata

Metadata

Assignees

Type

No type

Projects

Status

Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions