triton.language.extra.cann.extension.fixpipe

triton.language.extra.cann.extension.fixpipe(src: tensor, dst: None | buffer = None, dma_mode: FixpipeDMAMode = FixpipeDMAMode.NZ2ND, dual_dst_mode: FixpipeDualDstMode = FixpipeDualDstMode.NO_DUAL, _semantic=None) → None | tensor

Directly store a tensor on L0C to a local buffer via fixpipe. Fixpipe is pipeline that performing data movement from L0C to other memory hierarchies. Currently support:

  • L0C to UB (for Ascend910_95 sereies)

参数:

  • src (tl.tensor) -- the source tensor, Must be located in the l0C memory region.

  • dst (bl.buffer | None) -- The destination buffer in UB. If None, an empty tensor will be created. Must be buffer type in UB if provided.

  • dma_mode (FixpipeDMAMode) -- DMA transfer mode, "nz2nd" enables NZ to ND layout transformation

  • dual_dst_mode (FixpipeDualDstMode) -- Dual destination mode for split operations

返回:

If dst is None, returns the created tensor; otherwise returns None.

返回类型:

tl.tensor | None