Skip to content

Commit

Permalink
[IFRT] add c-bindings for "Held" PjRt classes (#751)
Browse files Browse the repository at this point in the history 
* [IFRT] add c-bindings for "Held" PjRt classes

* add `ifrt_client_assemble_array_from_single_shards`

* remove some legacy commented methods

* add `ifrt_compile`

* fix program type

* readd `ifrt_pjrt_compile` and make `ifrt_compile` use default compile options

* fix `ifrt_compile`

* get `Device` from `Client`

---------

Co-authored-by: Avik Pal <avikpal@mit.edu>
  • Loading branch information
@mofeing @avik-pal
mofeing and avik-pal authored Feb 19, 2025
1 parent c817f8a commit acaabc4
Show file tree
Hide file tree
Showing 2 changed files with 326 additions and 34 deletions.
Loading

0 comments on commit acaabc4

Please sign in to comment.