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

[tk] Extend type system and indexing sufficient to generate kernel functions. #219

Merged
merged 3 commits into from
Dec 4, 2023

Conversation

stellaraccident
Copy link
Contributor

  • More boilerplate for dependent types in the KernelBuffer hierarchy:
    • Adds usage (input, output, temporary)
    • Adds element_type and maps to torch.dtype
  • Adds a stub IndexingContext which can only handle constant symbol bindings right now.
  • Adds a vector_codegen emitter and gets it to the point that it can emit function definitions.
  • Next step for the iota kernel is to add grid bindings to the signature, emit the program_id operator and finish modeling the store from local to global memory. This can probably be done without more FX level passes, which would be needed for anything more complicated.

@stellaraccident stellaraccident merged commit fac744b into main Dec 4, 2023
3 checks passed
@stellaraccident stellaraccident deleted the tk3 branch December 4, 2023 06:58
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.

1 participant