tf-encrypted / moose

Secure distributed dataflow framework for encrypted machine learning and data processing
Apache License 2.0
58 stars 16 forks source link

Elk memory: Investigate improvements around `add_operation` #975

Open mortendahl opened 2 years ago

mortendahl commented 2 years ago

SymbolicSession::add_operation is (perhaps unsurprisingly) responsible for a lot of memory allocations. This issue is about investigate what, if anything, can be done to optimize this.

mortendahl commented 2 years ago

Maybe one idea is to let SymbolicSession generate an IndexedComputation instead of a NamedComputation: https://github.com/tf-encrypted/runtime/issues/977.