Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Extend TensorComputeOp to allow scalar inputs (#2606). #3300

Merged
merged 1 commit into from
Jun 22, 2019

Conversation

jdavies-huawei
Copy link
Contributor

@jdavies-huawei jdavies-huawei commented Jun 6, 2019

This is the second pull request for this change. In response to comments from @ZihengJiang, the scalar inputs can now be passed to the TensorIntrin without using a named parameter.

Before:

C = tvm.compute((n, n), lambda i: tfunc(A[i, 0:n], scalar_inputs=(i*i)), name='C')

Now:

C = tvm.compute((n, n), lambda i: tfunc(A[i, 0:n], i*i), name='C')

@jdavies-huawei
Copy link
Contributor Author

@ZihengJiang can you please review?

@jdavies-huawei
Copy link
Contributor Author

jdavies-huawei commented Jun 13, 2019

@kevinthesun can you add me to the slack channel? I requested on TVM discuss https://discuss.tvm.ai/t/request-for-invite-to-the-slack-channel/2888 3 days ago, but I haven't gotten a response yet.

@ZihengJiang ZihengJiang self-assigned this Jun 14, 2019
Copy link
Contributor

@ZihengJiang ZihengJiang left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants