mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
small typo fix (#2799)
This commit is contained in:
parent
3f8c73593d
commit
aee18c8860
@ -269,7 +269,7 @@ For parallel programming of multiple accelerators, like multiple GPUs, use
|
||||
With `pmap` you write single-program multiple-data (SPMD) programs, including
|
||||
fast parallel collective communication operations. Applying `pmap` will mean
|
||||
that the function you write is compiled by XLA (similarly to `jit`), then
|
||||
replicated and executed in parallel accross devices.
|
||||
replicated and executed in parallel across devices.
|
||||
|
||||
Here's an example on an 8-GPU machine:
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user