frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[reland][inductor] Register an op for mm_plus_mm (#105153)

Browse Source 
Summary: Reland https://github.com/pytorch/pytorch/pull/104835 after fixing internal build issues

Test Plan: CI

Differential Revision: D47442849

Pull Request resolved: https://github.com/pytorch/pytorch/pull/105153
Approved by: https://github.com/clee2000
This commit is contained in:
Bin Bao
2023-07-14 14:35:29 +00:00
committed by PyTorch MergeBot
parent c099b7e07a
commit 528ab477ce
8 changed files with 86 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -1160,6 +1160,7 @@ def main():
'include/torch/csrc/distributed/autograd/functions/*.h',
'include/torch/csrc/distributed/autograd/rpc_messages/*.h',
'include/torch/csrc/dynamo/eval_frame.h',
'include/torch/csrc/inductor/*.h',
'include/torch/csrc/jit/*.h',
'include/torch/csrc/jit/backends/*.h',
'include/torch/csrc/jit/generated/*.h',