src/pqc_gpu_driver/__init__.py
| 1 | """PQC-Hardened GPU Driver - encrypted CPU-GPU tensor transfers with ML-KEM + AES-GCM.""" |
| 2 | |
| 3 | from pqc_gpu_driver.backends.base import GPUBackend |
| 4 | from pqc_gpu_driver.backends.cuda import CUDABackend |
| 5 | from pqc_gpu_driver.backends.memory import InMemoryBackend |
| 6 | from pqc_gpu_driver.backends.rocm import ROCmBackend |
| 7 | from pqc_gpu_driver.channel import ChannelSession, establish_channel |
| 8 | from pqc_gpu_driver.driver_attest import ( |
| 9 | DriverAttestation, |
| 10 | DriverAttestationVerifier, |
| 11 | DriverAttester, |
| 12 | DriverModule, |
| 13 | ) |
| 14 | from pqc_gpu_driver.errors import ( |
| 15 | BackendError, |
| 16 | ChannelEstablishmentError, |
| 17 | ChannelExpiredError, |
| 18 | DecryptionError, |
| 19 | DriverAttestationError, |
| 20 | GPUDriverError, |
| 21 | NonceReplayError, |
| 22 | ) |
| 23 | from pqc_gpu_driver.tensor import EncryptedTensor, TensorMetadata |
| 24 | |
| 25 | __version__ = "0.1.0" |
| 26 | __all__ = [ |
| 27 | "EncryptedTensor", |
| 28 | "TensorMetadata", |
| 29 | "ChannelSession", |
| 30 | "establish_channel", |
| 31 | "DriverModule", |
| 32 | "DriverAttestation", |
| 33 | "DriverAttester", |
| 34 | "DriverAttestationVerifier", |
| 35 | "GPUBackend", |
| 36 | "InMemoryBackend", |
| 37 | "CUDABackend", |
| 38 | "ROCmBackend", |
| 39 | "GPUDriverError", |
| 40 | "ChannelEstablishmentError", |
| 41 | "ChannelExpiredError", |
| 42 | "NonceReplayError", |
| 43 | "DecryptionError", |
| 44 | "DriverAttestationError", |
| 45 | "BackendError", |
| 46 | ] |
| 47 | |