Script.tcgen05.load

Script.tcgen05.load

Script.tcgen05.load(tensor)[source]

Load data from tensor memory into registers.

Copies the contents of a 2D tensor memory tensor into a register tensor.

Parameters:

tensor (TMemoryTensor) – The source tensor memory tensor. Must be 2D.

Returns:

ret – A register tensor containing the loaded data.

Return type:

RegisterTensor

Notes

  • Thread group: Must be executed by a warp-aligned thread group.

  • Hardware: Requires compute capability 10.0+ (sm_100).

  • PTX: tcgen05.ld.sync.aligned

Contents