Update task_extraction.cc

apache · May 31, 2022 · 16a8851 · 16a8851
1 parent 33c4381
commit 16a8851
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/relay/backend/task_extraction.cc b/src/relay/backend/task_extraction.cc
@@ -37,6 +37,7 @@ bool DefaultTaskFilter(const Array<te::Tensor>& args) {
   std::unordered_set<const TensorNode*> visited;
   for (const Tensor& v : args) {
     for (const PrimExpr& e : v->shape) {
+      // Dynamic shape is not supported for now
       if (!e->IsInstance<IntImmNode>()) {
         return false;
       }