tf-encrypted / moose

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

Kernel framework rework to improve rust compilation time #1024

Closed rdragos closed 2 years ago

rdragos commented 2 years ago

Replaced old macro logic with less code generated by generating the kernel execution code inside a function rather than the macro.

When running cargo llvm-lines inside the moose directory we were getting before this PR:

    1   Lines            Copies         Function name                                                                                                                               
    2   -----            ------         -------------                                                                                                                               
    3   18243700 (100%)  613636 (100%)  (TOTAL) 

whereas now we get

    1   Lines           Copies         Function name                                                                                                                                                                                                                                                                                                                       
    2   -----           ------         -------------                                                                                                                                                                                                                                                                                                                       
    3   5478783 (100%)  111036 (100%)  (TOTAL)