[TOPI] check empty array of x86 injective's iters (#15513)

wrongtest-intellif · wrongtest · web-flow · commit 907b29e544bd · 2023-08-09T15:24:45.000-07:00
check empty array of x86 injective's iters

Co-authored-by: wrongtest &lt;wrongtest@gmail.com&gt;
diff --git a/include/tvm/topi/x86/injective.h b/include/tvm/topi/x86/injective.h
@@ -51,7 +51,7 @@ inline Schedule schedule_injective_from_existing(Schedule sch, const Tensor& out
     auto c = axis[1];
     auto fused = detail::Fuse(sch[out], {n, c});  // for nhwc layout, fuse n and h
     sch[out].parallel(fused);
-  } else {
+  } else if (!axis.empty()) {
     sch[out].parallel(axis[0]);
   }
   return sch;

Original file line number	Diff line number	Diff line change
`@@ -51,7 +51,7 @@ inline Schedule schedule_injective_from_existing(Schedule sch, const Tensor& out`
`51`	`51`	`auto c = axis[1];`
`52`	`52`	`auto fused = detail::Fuse(sch[out], {n, c}); // for nhwc layout, fuse n and h`
`53`	`53`	`sch[out].parallel(fused);`
`54`		`- } else {`
	`54`	`+ } else if (!axis.empty()) {`
`55`	`55`	`sch[out].parallel(axis[0]);`
`56`	`56`	`}`
`57`	`57`	`return sch;`