is_cuda
MatrixMultiplicationKey.is_cuda
- 행렬 곱셈 키가 GPU에 저장되어 있으면
True이고, 그렇지 않으면 False입니다.
from desilofhe import Engine
gpu_engine = Engine(slot_count=64, mode="gpu")
gpu_secret_key = gpu_engine.create_secret_key()
gpu_matrix_multiplication_key = gpu_engine.create_matrix_multiplication_key(
gpu_secret_key
)
assert gpu_matrix_multiplication_key.is_cuda
cpu_engine = Engine(slot_count=64, mode="cpu")
cpu_secret_key = cpu_engine.create_secret_key()
cpu_matrix_multiplication_key = cpu_engine.create_matrix_multiplication_key(
cpu_secret_key
)
assert not cpu_matrix_multiplication_key.is_cuda