Add Textract OP

[bingmeiyou]

No description provided.

[Zhendong404]

这里和下面的mat2right缺少transpose支持

template <typename DstTileData, typename SrcTileData>
PTO_INTERNAL void TExtractToLeft(DstTileData &dst, SrcTileData &src, uint16_t indexRow, uint16_t indexCol)
{
    static_assert((SrcTileData::SFractal == SLayout::ColMajor && SrcTileData::isRowMajor) ||
                      (SrcTileData::SFractal == SLayout::RowMajor && !SrcTileData::isRowMajor) ||
                      (SrcTileData::Rows == 1 && SrcTileData::isRowMajor),
                  "TExtract: SrcTile Invalid Fractal");
    static_assert(DstTileData::SFractal == SLayout::RowMajor && !DstTileData::isRowMajor,
                  "TExtract: DstTile Invalid Fractal");
    constexpr bool isFp4Type = std::is_same<typename SrcTileData::DType, float4_e2m1x2_t>::value ||
                               std::is_same<typename SrcTileData::DType, float4_e1m2x2_t>::value;
    if constexpr (SrcTileData::Rows == 1 && SrcTileData::isRowMajor) {
        TExtractToAVector<DstTileData, SrcTileData, isFp4Type>(dst.data(), src.data(), indexRow, indexCol,
                                                               dst.GetValidCol());
    } else if constexpr (DstTileData::SFractal == SrcTileData::SFractal) {
        if constexpr (DstTileData::Compact == CompactMode::Normal) {
            TExtractToACompact<DstTileData, SrcTileData, isFp4Type>(dst.data(), src.data(), indexRow, indexCol,
                                                                    dst.GetValidRow(), dst.GetValidCol());
        } else {
            TExtractToA<DstTileData, SrcTileData, false, isFp4Type>(dst.data(), src.data(), indexRow, indexCol);
        }
    } else {
        if constexpr (DstTileData::Compact == CompactMode::Normal || sizeof(typename SrcTileData::DType) == 1) {
            TExtractToATransCompact<DstTileData, SrcTileData, isFp4Type>(dst.data(), src.data(), indexRow, indexCol,
                                                                         dst.GetValidRow(), dst.GetValidCol());
        } else {
            TExtractToA<DstTileData, SrcTileData, true, isFp4Type>(dst.data(), src.data(), indexRow, indexCol);
        }
    }
}

[Zhendong404]

index_row/index_col参数没有被使用，逻辑和C++实现没对上

[bingmeiyou]

问题分析

您指出的问题完全正确。我们的 Python 模板接收了 index_row, index_col 参数但没传给底层指令。经排查，PTO-ISA 实际上是支持 start-position 的，只是我们用错了层级。

完整的调用链路

TileLang DSL 模板 (textract_template.py)
  → pto.mte_l1_l0a(src, dst, m, k)     ← 无 start-position 参数
    → VPTOExpandWrapperOps lowering
      → load_cbuf_to_ca(src, dst, mStart, kStart, mStep, kStep, ...)  ← 有 m_start/k_start 参数！

关键发现：

1. mte_l1_l0a/l0b（DSL 层 API）没有 start-position 参数（VPTOOps.td:2669-2687 只有 source, destination, m, k, transpose）
2. 但 load_cbuf_to_ca（硬件层指令）有 m_start, k_start 参数（VPTOOps.td:1142-1164）
3. VPTOExpandWrapperOps.cpp:772 的 deriveLoadCbufToCaControl 函数硬编码 mStart=0, kStart=0：

   return LoadCbufToCbControl{zero, zero, mStep, kStep, stride, stride};
   //                      mStart=0  kStart=0

4. C++ Cube API (TExtractToACompact) 传 indexRow, indexCol 是通过硬件层 API（类似 load_cbuf_to_ca），而不是通过 mte_l1_l0a 这层

为什么没对上

C++ 实现直接调用硬件层 API load_cbuf_to_ca(mStart=indexRow, kStart=indexCol)，而我们用的是 DSL 层 mte_l1_l0a，它是一个 wrapper op，expand 后才变成 load_cbuf_to_ca。这个 wrapper op 目前没有传递 start-position 的通道，导致 expand 阶段的 deriveLoadCbufToCaControl 硬编码为 0。

结论：不是 PTO-ISA 不支持，是我们用了不完整的 wrapper op。硬件层 load_cbuf_to_ca/cb 已经有 m_start/k_start，需要让 mte_l1_l0a/l0b wrapper 把 start-position 透传下去。

Fix-pipe 路径（Acc → Mat）：mte_l0c_l1

同样的情况：

• mte_l0c_l1（wrapper）没有 start-position
• copy_matrix_cc_to_cbuf（硬件层）也没有显式的 start-position 参数（它通过 xm/xt packed 配置控制，整个 tile 从 L0C 提取是隐式的）
• C++ 的 FP 路径实际上也不使用 indexRow/indexCol 做偏移

需要的修改

方案 A（推荐）：为 mte_l1_l0a/l0b 增加 start_position keyword

让 mte_l1_l0a/l0b wrapper op 增加 start_row/start_col keyword 参数（类似 transpose keyword），然后在 VPTOExpandWrapperOps.cpp 的 deriveLoadCbufToCaControl 中使用这些值替代硬编码的 zero。

改动点：

1. VPTOOps.td — MteL1L0aOp/MteL1L0bOp 增加 Optional<I64>:$start_row 和 Optional<I64>:$start_col 参数（默认值 0）
2. semantic.py — _analyze_mte_stage_load 增加 start_row/start_col keyword 支持
3. frontend_ast.py — "mte_l1_l0a": frozenset({"transpose", "start_row", "start_col"})
4. VPTOExpandWrapperOps.cpp — deriveLoadCbufToCaControl 接收 start_row/start_col 参数，传给 LoadCbufToCaOp 的 m_start/k_start
5. textract_template.py — 模板体中传 start_row=index_row, start_col=index_col，移除 constraint 中 !=0 的检查

方案 B（绕路）：直接在模板中使用 load_cbuf_to_ca

不用 mte_l1_l0a wrapper，直接在 textract 模板中用 pto.load_cbuf_to_ca(src, dst, index_row, index_col, mStep, kStep, srcStride, dstStride)。

问题：需要手动计算 mStep/kStep/stride（目前 deriveLoadCbufToCaControl 做的事），而且 load_cbuf_to_ca 不是 TileLang DSL v1 的 public API（不在 semantic.py 的 surface 列表中）。

推荐方案 A，因为它对齐了 transpose 的设计模式（wrapper op 透传硬件配置），且改动集中可控。

[Zhendong404]

这些针对不同数据类型的lit用例可以考虑合并一下