chore: debugging fhe inference in gh action

zama-ai · Jun 4, 2024 · 746e5c2 · 746e5c2
1 parent 364dc5a
commit 746e5c2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/concrete/ml/common/preprocessors.py b/src/concrete/ml/common/preprocessors.py
@@ -1694,7 +1694,7 @@ def apply(self, graph: Graph) -> None:
             assert isinstance(lsbs_to_remove, int)
             assert isinstance(max_acc_size, int)
 
-            print(f"{number_of_tlus=}, {lsbs_to_remove=} {coefficients.shape=}, {offsets.shape=}")
+            # print(f"{number_of_tlus=}, {lsbs_to_remove=} {coefficients.shape=}, {offsets.shape=}")
             if number_of_tlus == -1:
                 # Don't do here for now
                 # print("SKIPPING NODE")

diff --git a/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py b/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py
@@ -18,7 +18,7 @@
 from concrete.ml.quantization import QuantizedModule
 from concrete.ml.torch.compile import compile_brevitas_qat_model
 
-SIMULATE_ONLY = True
+SIMULATE_ONLY = False
 CURRENT_DIR = Path(__file__).resolve().parent
 KEYGEN_CACHE_DIR = CURRENT_DIR.joinpath(".keycache")
 
@@ -128,11 +128,11 @@ def wrapper(*args, **kwargs):
 open("cifar10.graph", "w").write(str(quantized_numpy_module.fhe_circuit))
 open("cifar10.mlir", "w").write(quantized_numpy_module.fhe_circuit.mlir)
 
-import sys
-
-if sys.platform == "darwin":
-    print("skipping fhe evaluation on darwin platform")
-    sys.exit(0)
+# import sys
+#
+# if sys.platform == "darwin":
+#     print("skipping fhe evaluation on darwin platform")
+#     sys.exit(0)
 
 if not SIMULATE_ONLY:
     # Key generation