search

alpa-projects / alpa

Training and serving large-scale neural networks with auto parallelization.
https://alpa.ai
Apache License 2.0
3.08k stars 357 forks source link

Add get mhlo module method in WarppedHlo #925

Closed vincentccc closed 1 year ago

vincentccc commented 1 year ago

format.sh

LGTM. Can you please format the file to pass the CI test? We have a format.sh script to help you do so.

Thanks, already done yet