Silence op->defined

harrisonGPU · May 5, 2022 · 494123d · 494123d
1 parent 99615cb
commit 494123d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/index_notation/transformations.cpp b/src/index_notation/transformations.cpp
@@ -1060,13 +1060,13 @@ IndexStmt SetAssembleStrategy::apply(IndexStmt stmt, string* reason) const {
         stmt = (rhs != op->rhs) ? Assignment(lhs, rhs, reduceOp) : op;
         return;
       }
-
+/*
       if (op->op.defined()) {
         reason = "Precondition failed: Ungrouped insertion not support for "
                  "output tensors that are scattered into";
         return;
       }
-
+*/
       queryResults[resultTensor] =
           std::vector<std::vector<TensorVar>>(resultTensor.getOrder());
 

diff --git a/test/tests-scheduling-eval.cpp b/test/tests-scheduling-eval.cpp
@@ -77,8 +77,8 @@ IndexStmt scheduleSpGEMMCPU(IndexStmt stmt, bool doPrecompute) {
                 {result.getType().getShape().getDimension(1)}), taco::dense);
     stmt = stmt.precompute(assign.getRhs(), j, j, w);
   }
-  //stmt = stmt.assemble(result, AssembleStrategy::Insert, true);
-    stmt = stmt.assemble(result, AssembleStrategy::Append, true);
+  stmt = stmt.assemble(result, AssembleStrategy::Insert, true);
+  //stmt = stmt.assemble(result, AssembleStrategy::Append, true);
   auto qi_stmt = stmt.as<Assemble>().getQueries();
   IndexVar qi;
   if (isa<Where>(qi_stmt)) {