Adds call to use inserTemporaries

tensor-compiler · May 30, 2019 · f4408d9 · f4408d9
1 parent 4f4ac27
commit f4408d9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/src/index_notation/transformations.cpp b/src/index_notation/transformations.cpp
@@ -654,7 +654,9 @@ static IndexStmt optimizeSpMM(IndexStmt stmt) {
   }
 
   // It's an SpMM statement so return an optimized SpMM statement
-  TensorVar w("w", Type(Float64, {Dimension()}), dense);
+  TensorVar w("w",
+              Type(Float64, {A.getType().getShape().getDimension(1)}),
+              dense);
   return forall(i,
                 where(forall(j,
                              A(i,j) = w(j)),

diff --git a/src/tensor.cpp b/src/tensor.cpp
@@ -461,6 +461,7 @@ void TensorBase::compile(bool assembleWhileCompute) {
     IndexStmt stmt = makeConcrete(assignment);
     string reason;
     stmt = reorderLoopsTopologically(stmt);
+    stmt = insertTemporaries(stmt);
     taco_uassert(stmt != IndexStmt()) << reason;
     stmt = parallelizeOuterLoop(stmt);
     content->assembleFunc = lower(stmt, "assemble", true, false);

diff --git a/tools/taco.cpp b/tools/taco.cpp
@@ -817,6 +817,7 @@ int main(int argc, char* argv[]) {
 
       string reason;
       stmt = reorderLoopsTopologically(stmt);
+      stmt = insertTemporaries(stmt);
       taco_uassert(stmt != IndexStmt()) << reason;
       stmt = parallelizeOuterLoop(stmt);
       compute = lower(stmt, "compute",  false, true);