Implement MatrixPhi support in compiler

facebookresearch / beanmachine

A library that allows for inference on probabilistic models

https://beanmachine.org/

MIT License

265 stars 49 forks source link

Implement MatrixPhi support in compiler #1787

Closed ericlippert closed 2 years ago

ericlippert commented 2 years ago

Summary: The compiler now "vectorizes" phi operators in the graph; if the input to a phi operator is a matrix then it generates the MatrixPhi node and ensures that its input is a real matrix.

Reviewed By: feynmanliang

Differential Revision: D39394979

facebook-github-bot commented 2 years ago

This pull request was exported from Phabricator. Differential Revision: D39394979