Closed MasterJH5574 closed 1 year ago
This PR contains two parts, with the first part being a pipeline prototype, and the second part being the a backend of Torch Dynamo using Relax.
The pipeline prototype contains the collection of pre-defined pipelines that optimizes and lower IRModule before passing to minimum build.
The Dynamo backend brings the mechanism to build PyTorch model using Relax compilation pipeline.
Full credit for the pipeline implementation goes to Tianqi, and full credit for the Dynamo backend goes to Bohan.
Co-authored-by: Tianqi Chen tianqi.tchen@gmail.com Co-authored-by: Bohan Hou 32121147+spectrometerHBH@users.noreply.github.com
This PR contains two parts, with the first part being a pipeline prototype, and the second part being the a backend of Torch Dynamo using Relax.
The pipeline prototype contains the collection of pre-defined pipelines that optimizes and lower IRModule before passing to minimum build.
The Dynamo backend brings the mechanism to build PyTorch model using Relax compilation pipeline.
Full credit for the pipeline implementation goes to Tianqi, and full credit for the Dynamo backend goes to Bohan.
Co-authored-by: Tianqi Chen tianqi.tchen@gmail.com Co-authored-by: Bohan Hou 32121147+spectrometerHBH@users.noreply.github.com