Script.atomic.shared_scatter_sub

Script.atomic.shared_scatter_sub(dst, *, dim, indices, values, sem='relaxed', scope='cta', output=None)

Scatter-sub into a shared tile along dim; lowered to atom.add with a negated value.

See shared_scatter_add() for the parameter description.

Notes

• Thread group: Can be executed by any sized thread group.

• Hardware: Requires compute capability 7.0+ (sm_70).

• PTX: atom.{sem}.{scope}.shared.add.s32 with a negated input.

Parameters:

• dst (SharedTensor)

• dim (int)

• indices (RegisterTensor)

• values (RegisterTensor)

• sem (str)

• scope (str)

• output (RegisterTensor | None)

Return type:

RegisterTensor | None