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

CenterPoint Backbone preprocessing optimization #83

Open
angry-crab opened this issue Dec 19, 2022 · 1 comment
Open

CenterPoint Backbone preprocessing optimization #83

angry-crab opened this issue Dec 19, 2022 · 1 comment
Labels
enhancement New feature or request

Comments

@angry-crab
Copy link
Collaborator

angry-crab commented Dec 19, 2022

The current implementation of scatter has some limitation.

  1. the GPU implementation hard coded iterator bindings which might not work for certain devices. For example, for OpenCL backend, if a GPU has only one dimension global work size.
        for j in T.thread_binding(0, 560, thread = "blockIdx.x"):
            for k in T.thread_binding(0, 560, thread = "blockIdx.y"):
                for i in T.thread_binding(0, 32, thread = "threadIdx.x"):
  1. There is no room for optimization because of hard code. Normally, we need to create schedule from IRModule and define optimization strategies.

  2. Need to create a optimization schedule and measure its performance.

@angry-crab angry-crab added the enhancement New feature or request label Dec 19, 2022
@angry-crab
Copy link
Collaborator Author

angry-crab commented Dec 19, 2022

1 can be solved by implementing scatter from a top level, ie TE or Relay.

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

No branches or pull requests

1 participant