[tvm] branch main updated: make injective ops's opt schedule applied to every output tensor (#11820)

[ma...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 1115fd9bc2 make injective ops's opt schedule applied to every output tensor (#11820)
1115fd9bc2 is described below

commit 1115fd9bc261619ffa0539746ae0aebc46232dc6
Author: Ivy Zhang <ya...@intel.com>
AuthorDate: Mon Jun 27 10:30:14 2022 +0800

    make injective ops's opt schedule applied to every output tensor (#11820)
---
 python/tvm/topi/x86/injective.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/python/tvm/topi/x86/injective.py b/python/tvm/topi/x86/injective.py
index 78893397ba..d197b50469 100644
--- a/python/tvm/topi/x86/injective.py
+++ b/python/tvm/topi/x86/injective.py
@@ -74,12 +74,11 @@ def schedule_injective(outs):
         The computation schedule for the op.
     """
     outs = [outs] if isinstance(outs, te.tensor.Tensor) else outs
-    x = outs[0]
     s = te.create_schedule([x.op for x in outs])
     te.schedule.AutoInlineInjective(s)
-
-    if not is_empty_shape(x.shape):
-        schedule_injective_from_existing(s, x)
+    for x in outs:
+        if not is_empty_shape(x.shape):
+            schedule_injective_from_existing(s, x)
     return s