don't run autoscheduling commands if manual schedule is provided in c…

…ommand line tool fixes #336
tensor-compiler · Dec 4, 2020 · 486cd9f · 486cd9f
1 parent 56c1e3e
commit 486cd9f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tools/taco.cpp b/tools/taco.cpp
@@ -929,8 +929,6 @@ int main(int argc, char* argv[]) {
   IndexStmt stmt =
       makeConcreteNotation(makeReductionNotation(tensor.getAssignment()));
   stmt = reorderLoopsTopologically(stmt);
-  stmt = insertTemporaries(stmt);
-  stmt = parallelizeOuterLoop(stmt);
 
   if (setSchedule) {
     stringstream scheduleStream; 
@@ -940,6 +938,10 @@ int main(int argc, char* argv[]) {
 
     cuda |= setSchedulingCommands(scheduleStream, parser, stmt);
   }
+  else {
+    stmt = insertTemporaries(stmt);
+    stmt = parallelizeOuterLoop(stmt);
+  }
 
   if (cuda) {
     if (!CUDA_BUILT && benchmark) {