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.

Sign up for GitHub

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

Jump to bottom

Investigate commonality between AOT Inductor/Triton/SPIRV Runner and refactor SYCL kernel launcher #2395

Open
kballeda opened this issue Oct 1, 2024 · 1 comment
Open

Investigate commonality between AOT Inductor/Triton/SPIRV Runner and refactor SYCL kernel launcher #2395

kballeda opened this issue Oct 1, 2024 · 1 comment
Assignees
@alexbaden
Labels
codegen: spirv enhancement New feature or request research
Milestone
0.3 [Triton] Lang...

Comments

@kballeda
Copy link
Contributor

kballeda commented Oct 1, 2024

Find commonality between AOT Inductor needs, SPIRV Runner needs, and Triton needs and refactor the launch code in driver.py to separate out the SYCL-only parts.

PyTorch AOT Inductor will also be using SYCL launch code for running Triton kernels outside of Triton
Reference: https://github.com/pytorch/pytorch/pull/135318/files#diff-2bbc58fcef7d498ae6d81ce8751f411d60ce5c4e8da9aa8dd95ccd29b64a1ba8
@kballeda kballeda self-assigned this Oct 1, 2024
@kballeda kballeda added codegen: spirv enhancement New feature or request labels Oct 1, 2024
@kballeda kballeda mentioned this issue Oct 1, 2024
Closed
@vlad-penkin
Copy link
Contributor

The preparation work for enabling AOT Inductor mode is in scope for

@alexbaden is this research task needed?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@alexbaden alexbaden

Labels
codegen: spirv enhancement New feature or request research
Projects
None yet
Milestone
0.3 [Triton] Language and Runtime
Development

No branches or pull requests
3 participants
@alexbaden @vlad-penkin @kballeda