Adds call to use inserTemporaries

Author: fredrikbk

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)),

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);

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);