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

Runtime support for RuntimeArgs on KernelDesc (Tensor Addr + Semaphores) #650

Merged
merged 1 commit into from
Sep 9, 2024

Conversation

kmabeeTT
Copy link
Contributor

@kmabeeTT kmabeeTT commented Sep 8, 2024

Closes #361 and Closes #267

  • 2 types of runtime_args today: RuntimeArgTensorAddress, RuntimeArgSemaphoreAddress where runtime will resolve tensor address or semaphore id, and call metal APIs SetRuntimeArgs() and CreateSemaphore()

  • Compiler support not included, so did some light testing with various dummy values for each type in flatbuffer via TTMetalToFlatbuffer.cpp changes, and tested assertions. For now, keep nullptr in .ttm for runtime_args and runtime_args_types

Copy link
Contributor

@nsmithtt nsmithtt left a comment

Choose a reason for hiding this comment

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

Minor, nit, otherwise looks great!

runtime/lib/ttmetal/command_queue.cpp Outdated Show resolved Hide resolved
…es) #361/#267

 - 2 types of runtime_args today: RuntimeArgTensorAddress,
   RuntimeArgSemaphoreAddress where runtime will resolve tensor address
   or semaphore id, and call metal APIs SetRuntimeArgs() and CreateSemaphore()

 - Compiler support not included, so did some light testing with various
   dummy values for each type in flatbuffer via TTMetalToFlatbuffer.cpp
   changes, and tested assertions. For now, keep nullptr in .ttm for
   runtime_args and runtime_args_types
@kmabeeTT kmabeeTT force-pushed the kmabee/issue_361_rt_args branch from 0d90c69 to 8c636ae Compare September 9, 2024 16:52
@kmabeeTT kmabeeTT merged commit 6bec67a into main Sep 9, 2024
11 checks passed
uazizTT pushed a commit that referenced this pull request Sep 12, 2024
…es) #361/#267 (#650)

- 2 types of runtime_args today: RuntimeArgTensorAddress,
   RuntimeArgSemaphoreAddress where runtime will resolve tensor address
   or semaphore id, and call metal APIs SetRuntimeArgs() and CreateSemaphore()

 - Compiler support not included, so did some light testing with various
   dummy values for each type in flatbuffer via TTMetalToFlatbuffer.cpp
   changes, and tested assertions. For now, keep nullptr in .ttm for
   runtime_args and runtime_args_types
@kmabeeTT kmabeeTT deleted the kmabee/issue_361_rt_args branch December 8, 2024 15:13
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.

Support Kernel Arguments Support metal Semaphores
2 participants