mirror of
https://github.com/zama-ai/tfhe-rs.git
synced 2026-04-28 03:01:21 -04: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