Forward XLA_FLAGS and TF_XLA_FLAGS in AOT compilation.

cms-ml / cmsml

Python Package of the CMS Machine Learning Group

https://cmsml.readthedocs.io

BSD 3-Clause "New" or "Revised" License

19 stars 6 forks source link

Closed riga closed 6 months ago

riga commented 6 months ago

This short PR aids the AOT process by forwarding XLA_FLAGS and TF_XLA_FLAGS to the compilation command to control various optimization settings.

riga commented 6 months ago

Thanks @valsdav !