Skip to content

Commit e8d511c

Browse files
committed
Changed createZeroInitTensor to createInitTensor with NULL
1 parent 6c4048c commit e8d511c

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
@@ -1706,8 +1706,8 @@ class ConvertAtenOuterOp : public OpConversionPattern<AtenOuterOp> {
17061706
// Create a zero-initialized tensor with shape [lhsDim, rhsDim]
17071707
Value zeroTensor = createZeroInitTensor(
17081708
rewriter, loc, ValueRange{lhsDim, rhsDim}, elementType);
1709-
1710-
// Set up affine indexing maps:
1709+
1710+
// Set up affine indexing maps:
17111711
// We create a 2D loop iteration space. For the lhs, we use the first index
17121712
// (i), for the rhs, the second index (j), and for the result, both (i, j).
17131713
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)