产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
获取LocalTensor所在的TPosition逻辑位置,支持TPosition为VECIN、VECOUT、VECCALC、A1、A2、B1、B2、CO1、CO2。
1 | __aicore__ inline int32_t GetPosition() const |
无
LocalTensor所在的TPosition逻辑位置。
无
1 2 3 4 5 6 7 8 9 | // 示例 AscendC::TPosition srcPos = (AscendC::TPosition)inputLocal.GetPosition(); if (srcPos == AscendC::TPosition::VECCALC) { // 处理逻辑1 } else if (srcPos == AscendC::TPosition::A1) { // 处理逻辑2 } else { // 处理逻辑3 } |