google / gematria

Machine learning for machine code.
Apache License 2.0
78 stars 12 forks source link

Get LLVM Binary paths in setup instead of init #247

Closed boomanaiden154 closed 1 month ago

boomanaiden154 commented 1 month ago

This patch moves the code getting the LLVM binary paths to the beam DoFn setup function rather than init. This is necessary as some implementations of get_llvm_binary_path require extracting the binary from an archive to a temporary directory on the local machine. Putting this in init does not guarantee that this happens on anything other than the machine running it (one) whereas running it in setup ensures the appropriate setup takes place for each worker instance.