jax.experimental.pallas.semaphore_read

Contents

jax.experimental.pallas.semaphore_read#

jax.experimental.pallas.semaphore_read(sem_or_view)[source]#

Reads the value of a semaphore.

Parameters:

sem_or_view – A Ref (or view) representing a semaphore.

Returns:

A scalar Array containing the value of the semaphore.

Return type:

jax_typing.Array