helion.language.store

helion.language.store(tensor, index, value, extra_mask=None)[source]

Store a value to a tensor using a list of indices.

This function is equivalent to tensor[index] = value but allows setting extra_mask= to mask elements beyond the default masking based on the hl.tile range.

Parameters:

  • tensor (Tensor) – The tensor to store to

  • index (list[object]) – The indices to use to index into the tensor

  • value (Tensor | SymInt | float) – The value to store

  • extra_mask (Tensor | None) – The extra mask (beyond automatic tile bounds masking) to apply to the tensor

Return type:

None

Returns:

None