ViewEngine

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√

功能说明

ViewEngine是Tensor API中的视图引擎，负责持有一个指针对象Iterator，通过Begin()返回这段视图的起始位置。在LocalTensor中，ViewEngine和Layout一起组成一个tensor的视图：

LocalTensor<TensorAttribute<ViewEngine<Iterator>, LayoutType>>

函数原型

ViewEngine定义如下：

template <typename Iterator>
struct ViewEngine
{
    using iterator    = Iterator;
    using reference   = typename IterRef<iterator>::type;
    using elementType = typename IterEle<iterator>::type;
    using valueType   = typename IterVal<iterator>::type;

    __aicore__ inline constexpr iterator const& Begin() const;
    __aicore__ inline constexpr iterator& Begin();
    __aicore__ inline constexpr ViewEngine(iterator storage = {});
};

成员函数说明

ViewEngine()

• 功能说明

  根据给定的指针/迭代器对象构造ViewEngine对象。

• 函数原型

  __aicore__ inline constexpr ViewEngine(iterator storage = {});
  
  

• 参数说明

• storage：视图起始位置对应的指针/迭代器对象。

• 返回值说明

  构造函数无返回值。

Begin()

• 功能说明

  获取构造ViewEngine对象时的起始位置指针/迭代器。

• 函数原型

  __aicore__ inline constexpr iterator const& Begin() const;
  
  __aicore__ inline constexpr iterator& Begin();
  
  

• 返回值说明

• 返回构造时保存的storage，即当前视图的起始指针/迭代器。

约束说明

传入的iterators必须是有效的随机访问迭代器，以确保能够正确访问和操作数据。

调用示例

  using namespace AscendC::Te;

  constexpr int32_t size = 256;
  __gm__ float data[size];
  auto ptr = MakeGMmemPtr(data);

  auto layout = MakeLayout(MakeShape(2, 2), MakeStride(2, 1));
  auto tensor = MakeTensor(ptr, layout);

  auto engine = tensor.Engine();
  auto begin = engine.Begin();

  float a = begin[0];
  float b = tensor[MakeCoord(1, 1)];