Skip to content

Commit

Permalink
chore(gpu): update new_multi_gpu() description
Browse files Browse the repository at this point in the history
  • Loading branch information
pdroalves authored and agnesLeroy committed Jan 14, 2025
1 parent 3d19986 commit 59b3159
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions tfhe/src/core_crypto/gpu/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,7 @@ unsafe impl Send for CudaStreams {}
unsafe impl Sync for CudaStreams {}

impl CudaStreams {
/// Create a new `CudaStreams` structure with as many GPUs as there are on the machine,
/// if they are connected via Nvlink. If the multiple GPUs on the machine are not connected
/// via Nvlink, this function will panic on the Cuda side.
/// Create a new `CudaStreams` structure with as many GPUs as there are on the machine
pub fn new_multi_gpu() -> Self {
let gpu_count = setup_multi_gpu();
let mut gpu_indexes = Vec::with_capacity(gpu_count as usize);
Expand Down

0 comments on commit 59b3159

Please sign in to comment.