is_cuda

PublicKeyB.is_cuda

Is True if the public key B is stored on the GPU, False otherwise.

• Output
  • boolean

from desilofhe import Engine

gpu_engine = Engine(use_multiparty=True, mode="gpu")
gpu_secret_key = gpu_engine.create_secret_key()
gpu_public_key_a = gpu_engine.create_public_key_a()
gpu_public_key_b = gpu_engine.create_public_key_b(
    gpu_secret_key, gpu_public_key_a
)

assert gpu_public_key_b.is_cuda

cpu_engine = Engine(use_multiparty=True, mode="cpu")
cpu_secret_key = cpu_engine.create_secret_key()
cpu_public_key_a = cpu_engine.create_public_key_a()
cpu_public_key_b = cpu_engine.create_public_key_b(
    cpu_secret_key, cpu_public_key_a
)

assert not cpu_public_key_b.is_cuda