to_cuda

Engine.to_cuda(data)
data가 CPU에 저장되어 있으면 GPU로 이동시킵니다. 이 메서드는 라이브러리의 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