mirror of
https://github.com/luxfhe/torus.git
synced 2026-10-30 17:07:26 +00:00
zeekay
b832ba2c39
Forked from upstream concrete (BSD-3-Clause-Clear, © 2024 ZAMA) at HEAD
ecb729bb. Modifications copyright © 2026 Lux Industries Inc.
Rebranding scope:
- Python module concrete → torus (across frontends/torus-python)
- Rust module concrete-* → torus-* (across frontends/torus-rust)
- MLIR dialect concretelang → toruslang (across compilers/torus-compiler)
- Compiler optimizer concrete-optimizer → torus-optimizer
- Backend concrete-cpu / concrete-cuda → torus-cpu / torus-cuda
- Wire protocol concrete-protocol → torus-protocol
- Parameter-curves concrete-security-curves-* → torus-security-curves-*
- Public symbols concretelib → toruslib, ConcreteLang → TorusLang
- URLs github.com/zama-ai/concrete → github.com/luxfhe/torus
- URLs docs.zama.ai/concrete → docs.luxfhe.com/torus
Legal compliance:
- LICENSE.txt retains original Copyright © 2024 ZAMA verbatim (§1-§2)
- LICENSE.txt adds modifications copyright © 2026 Lux Industries Inc.
- §3 endorsement clause: Torus is marketed exclusively as a Lux product;
no Zama name used in README, docs, or marketing
- NOTICE file provides factual attribution per legal best practice
(not endorsement; required by upstream BSD-3-Clause-Clear)
README + NOTICE rewritten from scratch with Lux brand:
- Quick-start in Python (torus.fhe)
- Architecture diagram showing the gen-1 vtbl integration path
(luxcpp/gpu → lux-private/gpu-kernels for CUDA/Metal/WebGPU)
- Scheme coverage table: TFHE (luxfi/fhe), CKKS/BGV/BFV (luxfi/lattice)
- Lux FHE family table: Torus / TorusVM / TorusEVM / TorusML / TorusNet
Known limitations (phase-2A scope):
- PyPI package torus-fhe not yet published; the torus Python module
is renamed structurally but pip-installable distribution is a
follow-up (Phase 2A.1)
- MLIR dialect rename is text-level; the compiler-internal LLVM
rebuild (concrete-compiler-internal-llvm-project submodule) is
structural and works; tablegen files renamed concrete→torus but the
generated .inc files in the source tree will regenerate at build
time
- backends/torus-cuda/implementation/ is an upstream submodule
pointer to the tfhe-rs source (not fetched in the staging clone);
the CUDA backend body is delivered out-of-tree
- ~100 test-fixture variable names like concrete_encoded_sample
retain "concrete" as a math term (vs symbolic); these are inside
tests/, not the public API; cleanup is a follow-up commit
- 295 Python `from concrete import ...` residuals in luxfhe/ml-sdk
consumer remain; that repo's import update is Phase 2B
This commit is the seed for the luxfhe/torus repo. Push as the first
commit to a fresh GitHub repo; do not re-import upstream Zama git
history.
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
import numpy as np
|
|
import pytest
|
|
import shutil
|
|
import tempfile
|
|
|
|
from torus.compiler import (
|
|
Compiler,
|
|
KeyType,
|
|
PrimitiveOperation,
|
|
lookup_runtime_lib,
|
|
Backend,
|
|
CompilationOptions,
|
|
MoreCircuitCompilationFeedback,
|
|
)
|
|
|
|
|
|
def test_statistics():
|
|
mlir = """
|
|
|
|
module {
|
|
func.func @main(%arg0: !FHE.eint<3>) -> !FHE.eint<6> {
|
|
%cst = arith.constant dense<[0, 1, 4, 9, 16, 25, 36, 49]> : tensor<8xi64>
|
|
%0 = "FHE.apply_lookup_table"(%arg0, %cst) : (!FHE.eint<3>, tensor<8xi64>) -> !FHE.eint<6>
|
|
return %0 : !FHE.eint<6>
|
|
}
|
|
}
|
|
|
|
""".strip()
|
|
|
|
with tempfile.TemporaryDirectory() as tmpdirname:
|
|
support = Compiler(str(tmpdirname), lookup_runtime_lib())
|
|
library = support.compile(mlir, CompilationOptions(Backend.CPU))
|
|
|
|
program_info = library.get_program_info()
|
|
program_compilation_feedback = library.get_program_compilation_feedback()
|
|
compilation_feedback = program_compilation_feedback.get_circuit_feedback("main")
|
|
|
|
pbs_count = MoreCircuitCompilationFeedback.count(
|
|
compilation_feedback,
|
|
operations={
|
|
PrimitiveOperation.PBS,
|
|
PrimitiveOperation.WOP_PBS,
|
|
},
|
|
)
|
|
assert pbs_count == 1
|
|
|
|
pbs_counts_per_parameter = MoreCircuitCompilationFeedback.count_per_parameter(
|
|
compilation_feedback,
|
|
operations={
|
|
PrimitiveOperation.PBS,
|
|
PrimitiveOperation.WOP_PBS,
|
|
},
|
|
key_types={KeyType.BOOTSTRAP},
|
|
program_info=program_info,
|
|
)
|
|
assert len(pbs_counts_per_parameter) == 1
|
|
assert pbs_counts_per_parameter[list(pbs_counts_per_parameter.keys())[0]] == 1
|
|
|
|
pbs_counts_per_tag = MoreCircuitCompilationFeedback.count_per_tag(
|
|
compilation_feedback,
|
|
operations={
|
|
PrimitiveOperation.PBS,
|
|
PrimitiveOperation.WOP_PBS,
|
|
},
|
|
)
|
|
assert pbs_counts_per_tag == {}
|
|
|
|
pbs_counts_per_tag_per_parameter = (
|
|
MoreCircuitCompilationFeedback.count_per_tag_per_parameter(
|
|
compilation_feedback,
|
|
operations={
|
|
PrimitiveOperation.PBS,
|
|
PrimitiveOperation.WOP_PBS,
|
|
},
|
|
key_types={KeyType.BOOTSTRAP},
|
|
program_info=program_info,
|
|
)
|
|
)
|
|
assert pbs_counts_per_tag_per_parameter == {}
|