Skip to content

Commit ddf1108

Browse files
committed
Changed createZeroInitTensor to createInitTensor with NULL
1 parent c7e3e05 commit ddf1108

File tree

2 files changed

+3
-3
lines changed
  • lib/Conversion/TorchToLinalg
  • projects/pt1/python/torch_mlir_e2e_test/test_suite

2 files changed

+3
-3
lines changed

lib/Conversion/TorchToLinalg/Linear.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1792,8 +1792,8 @@ class ConvertAtenOuterOp : public OpConversionPattern<AtenOuterOp> {
17921792
// Create a zero-initialized tensor with shape [lhsDim, rhsDim]
17931793
Value zeroTensor = createZeroInitTensor(
17941794
rewriter, loc, ValueRange{lhsDim, rhsDim}, elementType);
1795-
1796-
// Set up affine indexing maps:
1795+
1796+
// Set up affine indexing maps:
17971797
// We create a 2D loop iteration space. For the lhs, we use the first index
17981798
// (i), for the rhs, the second index (j), and for the result, both (i, j).
17991799
AffineMap mapLhs =

projects/pt1/python/torch_mlir_e2e_test/test_suite/matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -941,4 +941,4 @@ def forward(self, lhs, rhs):
941941

942942
@register_test_case(module_factory=lambda: AtenOuter())
943943
def AtenOuter_basic(module, tu: TestUtils):
944-
module.forward(tu.rand(3), tu.rand(3))
944+
module.forward(tu.rand(3), tu.rand(3))

0 commit comments

Comments
 (0)