This is an automated email from the ASF dual-hosted git repository.
tqchen 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 73b01eec4a Fixing workload comment (#16662)
73b01eec4a is described below
commit 73b01eec4ab718d88f9c7a041b71ad8da68537c6
Author: Thais Camacho <[email protected]>
AuthorDate: Sun Mar 3 11:18:14 2024 -0300
Fixing workload comment (#16662)
Fixing workload comment.
---
python/tvm/auto_scheduler/workload_registry.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/auto_scheduler/workload_registry.py
b/python/tvm/auto_scheduler/workload_registry.py
index 117b6401b5..62ba2245b0 100644
--- a/python/tvm/auto_scheduler/workload_registry.py
+++ b/python/tvm/auto_scheduler/workload_registry.py
@@ -77,7 +77,7 @@ def register_workload(func_name, f=None, override=False):
A = te.placeholder((N, K), name='A')
B = te.placeholder((K, M), name='B')
k = te.reduce_axis((0, K), name='k')
- C = te.compute((N, M), lambda i, j: tvm.sum(A[i][k] * B[k][j],
axis=[k]), name='C')
+ C = te.compute((N, M), lambda i, j: te.sum(A[i][k] * B[k][j],
axis=[k]), name='C')
return [A, B, C]
"""
global WORKLOAD_FUNC_REGISTRY
