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

Conditionally copy back and output 1 or more tensors in SPIRVRunner #2569

Open
alexbaden opened this issue Oct 24, 2024 · 0 comments
Open

Conditionally copy back and output 1 or more tensors in SPIRVRunner #2569

alexbaden opened this issue Oct 24, 2024 · 0 comments
Assignees
Labels
enhancement New feature or request

Comments

@alexbaden
Copy link
Contributor

Currently SPIRVRunner assumes the user always wants to copy back 1 and only 1 output. But, there is no requirement that Triton kernels have a single output - in fact, input and output is just a matter of whether the kernel calls load or store on a pointer. It is conceivable that a user may want to copy back all output buffers, or copy none at all. We should provide that flexibility.

@vlad-penkin vlad-penkin added the enhancement New feature or request label Oct 28, 2024
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

3 participants