iree icon indicating copy to clipboard operation
iree copied to clipboard

[Epic] Add support for `mhlo.custom_call`

Open rsuderman opened this issue 2 years ago • 4 comments

There are some operations that we will likely support via library calls (e.g. QR, Eigh, etc). We should add support for a method to call out to related libraries to execute this solvers as they may not map well to the linalg code generation system.

(Everything below is EDIT by @vmurali )

This is the initial design document for this feature.

  • [x] #13172
  • [x] #13162
  • [ ] #13164
  • [x] https://github.com/google/jax/issues/15357
  • [ ] #13163
  • [ ] #13165
  • [ ] #13166
  • [ ] #13167
  • [ ] Profit

rsuderman avatar Feb 27 '23 23:02 rsuderman

@vmurali Can you help update this Epic?

allieculp avatar May 04 '23 18:05 allieculp

Finished first task and trying to finish the second task.

vmurali avatar May 04 '23 18:05 vmurali

(Updated priority.)

aaron-schneider avatar May 23 '23 17:05 aaron-schneider

@rsuderman Cleaning up issues - is this still active/relevant?

allieculp avatar Aug 29 '23 15:08 allieculp