We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e56c5e1 commit 91aff66Copy full SHA for 91aff66
python/tvm/auto_scheduler/workload_registry.py
@@ -77,7 +77,7 @@ def matmul(N, M, K):
77
A = te.placeholder((N, K), name='A')
78
B = te.placeholder((K, M), name='B')
79
k = te.reduce_axis((0, K), name='k')
80
- 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')
81
return [A, B, C]
82
"""
83
global WORKLOAD_FUNC_REGISTRY
0 commit comments