llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Integrate llvm-project at a8de667af092c9b4b3b4a95827a521602ebf14ed. 

* Requires patch https://reviews.llvm.org/D108527
pull/286/head
Stella Laurenzo 2021-08-22 15:21:35 -07:00 committed by Stella Laurenzo
parent 80ff744c56
commit 32f56c67f4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
external/llvm-project vendored

@ -1 +1 @@
Subproject commit 2d9759c7902c5cbc9a7e3ab623321d5578d51687 Subproject commit a8de667af092c9b4b3b4a95827a521602ebf14ed

2
lib/Conversion/TorchToLinalg/TorchToLinalg.cpp
View File

@ -351,7 +351,7 @@ public:
auto dilationAttr = rewriter.getI64VectorAttr(dilationInts); auto dilationAttr = rewriter.getI64VectorAttr(dilationInts);
Value conv2d = Value conv2d =
rewriter rewriter
.create<linalg::Conv2DNchwOp>( .create<linalg::Conv2DNchwFchwOp>(
loc, initTensor0.getType(), ValueRange{paddedInput, weight}, loc, initTensor0.getType(), ValueRange{paddedInput, weight},
initTensor0, stridesAttr, dilationAttr) initTensor0, stridesAttr, dilationAttr)
.getResult(0); .getResult(0);