Skip to content
This repository has been archived by the owner on Dec 18, 2023. It is now read-only.

BMInference API to compile and run BM models from Beanstalk #1801

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

BMInference API to compile and run BM models from Beanstalk #1801

wants to merge 1 commit into from

Commits on Nov 1, 2022

  1. BMInference API to compile and run BM models from Beanstalk (facebook… 

    …research#1801)

Summary:
Pull Request resolved: facebookresearch#1801

We currently have `BMGInference`, that takes a BM python program produces a valid BMG graph and runs inference on the BMG C++ backend. In the same vein, this diff implements a `BMInference` API that takes a BM python program, optimizes it (currently uses the same optimization path as BMG) and runs BM inference methods to compute the samples. This allows us to separate the type of optimizations each backend needs and allows us to test end-to-end with an inference method.

Differential Revision: D40853055

fbshipit-source-id: 8b5ce58385c5fe17e0a688e428261f3276e9b03b
    @AishwaryaSivaraman @facebook-github-bot
    AishwaryaSivaraman authored and facebook-github-bot committed Nov 1, 2022
    Configuration menu
    Copy the full SHA
    86990bc View commit details
    Browse the repository at this point in the history