Skip to content

Commit 91aff66

Browse files
committed
Fixing workload comment.
1 parent e56c5e1 commit 91aff66

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/auto_scheduler/workload_registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def matmul(N, M, K):
7777
A = te.placeholder((N, K), name='A')
7878
B = te.placeholder((K, M), name='B')
7979
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')
80+
C = te.compute((N, M), lambda i, j: te.sum(A[i][k] * B[k][j], axis=[k]), name='C')
8181
return [A, B, C]
8282
"""
8383
global WORKLOAD_FUNC_REGISTRY

0 commit comments

Comments
 (0)