How to compile TensorFlow code with our desired protocols?

[zwxandy]

Hi, I am very interested in your work!
Recently, I have a question about compiling TensorFlow code to cpp code. If I directly generate cpp code with the method provided by "Athos", I will get the cpp code that invokes protocols in "SCI/src/library_fixed_uniform.cpp" by default.
So, I wonder how I can compile TF code to cpp code with invoking protocols in "SCI/src/library_fixed_uniform_cheetah.cpp" that you've added for Cheetah. Which part of the code should I modify when I convert the code.
I would appreciate it very much if you could help me with this question! Thanks a lot!

[fionser]

• To invoke our code, we add a flag in the CMake so the whole project is compiled using Cheetah's code instead of the origin SCI's code.
• For example, this we create an extra library that compiles "cheetah.cpp "
• That is exactly how we plugin the Cheetah's code into SCI's framework and use the EzPC's compiler (i.e. Athos)

[fionser]

Basically, we keep to use the same API "library_fixed_uniform.h" which defines the functions used by the Compiler.
But we just switch to Cheetah's implementation during the compile time.

[zwxandy]

• To invoke our code, we add a flag in the CMake so the whole project is compiled using Cheetah's code instead of the origin SCI's code.
• For example, this we create an extra library that compiles "cheetah.cpp "
• That is exactly how we plugin the Cheetah's code into SCI's framework and use the EzPC's compiler (i.e. Athos)

Thank you so much!
So, the point of the modification is the CMake file (1. include library_fixed_uniform_cheetah.cpp protocols; 2. add a flag to choose a certain protocol when compiling).
I notice that library_fixed_uniform_cheetah.cpp uses the same function names as library_fixed_uniform.cpp and "library_fixed_uniform.h" (e.g., Conv2DWrapper). So, do you simply use the statements like "#if !USE_CHEETAH" to make sure the code is compiled with certain protocols?

[fionser]

Yes.