Skip to content

[TIR] Expose TVM Backend API-related Builtins and Misc#12431

Closed
cyx-6 wants to merge 1 commit intoapache:mainfrom
cyx-6:upstream-a1-7
Closed

[TIR] Expose TVM Backend API-related Builtins and Misc#12431
cyx-6 wants to merge 1 commit intoapache:mainfrom
cyx-6:upstream-a1-7

Conversation

@cyx-6
Copy link
Copy Markdown
Contributor

@cyx-6 cyx-6 commented Aug 15, 2022

op list:
tvm_thread_allreduce
tvm_access_ptr
tvm_throw_last_error
TVMBackendAllocWorkspace
TVMBackendFreeWorkspace
Used and tested for future TVMScript parser

Co-Authored-By: yongwww <55wuyong@163.com>

cc @junrushao1994

@junrushao junrushao changed the title [TIR] Expose TVM-builtins and runtime ops to python [TIR] Expose TVM Backend API-related Builtins Aug 15, 2022
@junrushao junrushao changed the title [TIR] Expose TVM Backend API-related Builtins [TIR] Expose TVM Backend API-related Builtins and Misc Aug 15, 2022
@github-actions github-actions bot requested a review from junrushao August 15, 2022 01:56
Copy link
Copy Markdown
Contributor

@leandron leandron left a comment

Choose a reason for hiding this comment

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

Hi @cyx-6 - thanks for the many PRs. I see that generally you're exposing TIR operators to Python so that TVMScript (which is described in an RFC). There are two things I'd like to improved in all there PRs:

  1. unit tests so that we can test the operations are actually accessible via Python
  2. a better commit message (currently set as update), that can be changed in the PR description. We would like to see the motivation for this change.

`op` list:
`tvm_thread_allreduce`
`tvm_access_ptr`
`tvm_throw_last_error`
`TVMBackendAllocWorkspace`
`TVMBackendFreeWorkspace`
@junrushao
Copy link
Copy Markdown
Member

See our discussion thread here. Let's do one PR at a time :-)

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.

3 participants