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.
125 lines
4.3 KiB
Python
125 lines
4.3 KiB
Python
"""
|
|
Tests of everything related to restrictions.
|
|
"""
|
|
|
|
import numpy as np
|
|
|
|
# pylint: disable=import-error
|
|
from mlir._mlir_libs._toruslang._compiler import (
|
|
KeysetInfo,
|
|
PartitionDefinition,
|
|
RangeRestriction,
|
|
)
|
|
|
|
from torus import fhe
|
|
|
|
# pylint: enable=import-error
|
|
|
|
|
|
# pylint: disable=missing-class-docstring, missing-function-docstring, no-self-argument, unused-variable, no-member, unused-argument, function-redefined, expression-not-assigned
|
|
# same disables for ruff:
|
|
# ruff: noqa: N805, E501, F841, ARG002, F811, B015
|
|
|
|
|
|
def test_range_restriction():
|
|
"""
|
|
Test that compiling a module works.
|
|
"""
|
|
|
|
@fhe.module()
|
|
class Module:
|
|
@fhe.function({"x": "encrypted"})
|
|
def inc(x):
|
|
return (x + 1) % 20
|
|
|
|
inputset = [np.random.randint(1, 20, size=()) for _ in range(100)]
|
|
range_restriction = RangeRestriction()
|
|
internal_lwe_dimension = 999
|
|
range_restriction.add_available_internal_lwe_dimension(internal_lwe_dimension)
|
|
glwe_log_polynomial_size = 12
|
|
range_restriction.add_available_glwe_log_polynomial_size(glwe_log_polynomial_size)
|
|
glwe_dimension = 2
|
|
range_restriction.add_available_glwe_dimension(glwe_dimension)
|
|
pbs_level_count = 3
|
|
range_restriction.add_available_pbs_level_count(pbs_level_count)
|
|
pbs_base_log = 11
|
|
range_restriction.add_available_pbs_base_log(pbs_base_log)
|
|
ks_level_count = 3
|
|
range_restriction.add_available_ks_level_count(ks_level_count)
|
|
ks_base_log = 6
|
|
range_restriction.add_available_ks_base_log(ks_base_log)
|
|
module = Module.compile({"inc": inputset}, range_restriction=range_restriction)
|
|
keyset_info = module.keys.specs.program_info.get_keyset_info()
|
|
assert keyset_info.bootstrap_keys()[0].polynomial_size() == 2**glwe_log_polynomial_size
|
|
assert keyset_info.bootstrap_keys()[0].input_lwe_dimension() == internal_lwe_dimension
|
|
assert keyset_info.bootstrap_keys()[0].glwe_dimension() == glwe_dimension
|
|
assert keyset_info.bootstrap_keys()[0].level() == pbs_level_count
|
|
assert keyset_info.bootstrap_keys()[0].base_log() == pbs_base_log
|
|
assert keyset_info.keyswitch_keys()[0].level() == ks_level_count
|
|
assert keyset_info.keyswitch_keys()[0].base_log() == ks_base_log
|
|
assert keyset_info.secret_keys()[0].dimension() == 2**glwe_log_polynomial_size * glwe_dimension
|
|
assert keyset_info.secret_keys()[1].dimension() == internal_lwe_dimension
|
|
|
|
|
|
def test_keyset_restriction():
|
|
"""
|
|
Test that compiling a module works.
|
|
"""
|
|
|
|
@fhe.module()
|
|
class Big:
|
|
@fhe.function({"x": "encrypted"})
|
|
def inc(x):
|
|
return (x + 1) % 200
|
|
|
|
big_inputset = [np.random.randint(1, 200, size=()) for _ in range(100)]
|
|
|
|
@fhe.module()
|
|
class Small:
|
|
@fhe.function({"x": "encrypted"})
|
|
def inc(x):
|
|
return (x + 1) % 20
|
|
|
|
small_inputset = [np.random.randint(1, 20, size=()) for _ in range(100)]
|
|
|
|
big_module = Big.compile(
|
|
{"inc": big_inputset},
|
|
)
|
|
big_keyset_info = big_module.keys.specs.program_info.get_keyset_info()
|
|
|
|
small_module = Small.compile(
|
|
{"inc": small_inputset},
|
|
)
|
|
small_keyset_info = small_module.keys.specs.program_info.get_keyset_info()
|
|
assert big_keyset_info != small_keyset_info
|
|
|
|
restriction = big_keyset_info.get_restriction()
|
|
restricted_module = Small.compile({"inc": small_inputset}, keyset_restriction=restriction)
|
|
restricted_keyset_info = restricted_module.keys.specs.program_info.get_keyset_info()
|
|
assert big_keyset_info == restricted_keyset_info
|
|
assert small_keyset_info != restricted_keyset_info
|
|
|
|
|
|
def test_generic_restriction():
|
|
"""
|
|
Test that compiling a module works.
|
|
"""
|
|
|
|
generic_keyset_info = KeysetInfo.generate_virtual(
|
|
[PartitionDefinition(8, 10.0), PartitionDefinition(10, 10000.0)], True
|
|
)
|
|
|
|
@fhe.module()
|
|
class Module:
|
|
@fhe.function({"x": "encrypted"})
|
|
def inc(x):
|
|
return (x + 1) % 200
|
|
|
|
inputset = [np.random.randint(1, 200, size=()) for _ in range(100)]
|
|
restricted_module = Module.compile(
|
|
{"inc": inputset},
|
|
keyset_restriction=generic_keyset_info.get_restriction(),
|
|
)
|
|
compiled_keyset_info = restricted_module.keys.specs.program_info.get_keyset_info()
|
|
assert all(k in generic_keyset_info.secret_keys() for k in compiled_keyset_info.secret_keys())
|