mirror of
https://github.com/zama-ai/tfhe-rs.git
synced 2026-01-10 23:28:11 -05:00
Only GPUs with peer access to GPU 0 can be used for this at the moment. Peer to peer copy is used if different GPUs are passed to memcpy_gpu_to_gpu A gpu offset is passed as new parameter to pbs and keyswitch to adjust the input/output index user per gpu. bsk and ksk are copied to all GPUs. The CI now tests & runs benchmarks on p3.8xlarge aws instances