jax.experimental.pallas.mosaic_gpu.SemaphoreSignal#

class jax.experimental.pallas.mosaic_gpu.SemaphoreSignal(ref: '_Ref', *, device_id: 'pallas_primitives.DeviceId | None', inc: 'int | jax.Array' = 1)[来源]#
参数:

  • ref (_Ref)

  • device_id (pallas_primitives.DeviceId | None)

  • inc (int | jax.Array)

__init__(ref, *, device_id, inc=1)#
参数:

  • ref (_Ref)

  • device_id (pallas_primitives.DeviceId | None)

  • inc (int | jax.Array)

返回类型:

无

方法

__init__(ref, *, device_id[, inc])

属性

inc

ref

device_id