is_cuda
EvaluationKey.is_cuda
- Is
True if the evaluation key is stored on the GPU, False otherwise.
from desilofhe import Engine
gpu_engine = Engine(bootstrap=True, mode="gpu")
gpu_secret_key = gpu_engine.create_secret_key()
gpu_evaluation_key = gpu_engine.create_evaluation_key(gpu_secret_key)
assert gpu_evaluation_key.is_cuda
cpu_engine = Engine(bootstrap=True, mode="cpu")
cpu_secret_key = cpu_engine.create_secret_key()
cpu_evaluation_key = cpu_engine.create_evaluation_key(cpu_secret_key)
assert not cpu_evaluation_key.is_cuda