to_cuda

Engine.to_cuda(data)

data가 CPU에 저장되어 있으면 GPU로 이동시킵니다. 엔진이 GPU 모드가 아니면 예외가 발생합니다.

• 인풋:
  • 엔진으로 생성된 모든 자료 구조체
• 아웃풋:
  • None

from desilofhe import Engine

engine = Engine(mode="gpu")
secret_key = engine.create_secret_key()

assert secret_key.is_cuda

engine.to_cpu(secret_key)

assert not secret_key.is_cuda

engine.to_cuda(secret_key)

assert secret_key.is_cuda