is_cuda
EvaluationKey.is_cuda
- 연산 키가 GPU에 저장되어 있으면
True이고, 그렇지 않으면 False입니다.
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