-
Notifications
You must be signed in to change notification settings - Fork 25.7k
Closed
Description
Currently a lot of mkldnn-specific functions and classes reside in _inductor/overrides.py under fairly generic names such as ConvUnary2d
pytorch/torch/_inductor/overrides.py
Line 84 in 3c637e8
| class ConvUnary2d(nn.Conv2d): |
fuse_unary pytorch/torch/_inductor/overrides.py
Line 699 in 3c637e8
| def fuse_unary(gm: torch.fx.GraphModule): |
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
desertfire
Metadata
Metadata
Assignees
Type
Projects
Status
Done