AI エンジン API では、多数の演算に対して演算子のオーバーロードがサポートされています。この機能を使用するには、ヘッダー ファイル aie_api/operators.hpp を含め、aie::operators 名前空間を使用します。
含まれている演算子は次のとおりです。
- 演算子
+ - 加算演算子。
aie::addと等価です。 - 演算子
+= - 加算代入演算子。
- 演算子
- - 否定演算子。
aie::negと等価です。 - 演算子
- - 減算演算子。
aie::subと等価です。 - 演算子
-= - 減算代入演算子。
- 演算子
!= - 不等号比較演算子。
aie::neqと等価です。 - 演算子
< - 小なり比較演算子。
aie::ltと等価です。 - 演算子
<= - 以下比較演算子。
aie::leと等価です。 - 演算子
== - 等号比較演算子。
aie::eqと等価です。 - 演算子
> - 大なり比較演算子。
aie::gtと等価です。 - 演算子
>= - 以上比較演算子。
aie::geと等価です。 - 演算子
<< - ビット単位左シフト演算子。
aie::upshiftと等価です。 - 演算子
>> - ビット単位右シフト演算子。
aie::downshiftと等価です。 - 演算子
& - ビット単位 AND 演算子。
aie::bit_andと等価です。 - 演算子
^ - ビット単位 XOR 演算子。
aie::bit_xorと等価です。 - 演算子
| - ビット単位 OR 演算子。
aie::bit_orと等価です。 - 演算子
~ - ビット単位 NEG 演算子。
aie::bit_notと等価です。
次に、演算子を使用したコード例を示します。
#include <aie_api/aie.hpp>
#include <aie_api/aie_adf.hpp>
#include <aie_api/utils.hpp>
#include <aie_api/operators.hpp>
using namespace aie::operators;
aie::vector<int32,8> va,vb;
aie::vector<int32,8> vadd=va+vb;
aie::vector<int32,8> vsub=va-vb;
// negation of each element
aie::vector<int32,8> vneg=-vb;
vadd+=vneg;
aie::print(va,true,"va=");
aie::print(vadd,true,"vadd=");
// compare each element of the vector correspondingly
auto msk_neq=(vadd!=va);
// mask bit equals 1 if vector element not equal
aie::print(msk_neq,true,"msk_neq=");
// bit not
auto vnot_va=~va;
// bit xor
auto vones=va ^ vnot_va;
aie::print(vones,true,"vones=");
// choose vadd if mask bit equals 0
auto vout=aie::select(vadd,vones,msk_neq);
aie::print(vout,true,"vout=");次に、上記のコードの出力例を示します。
va=9 10 11 12 13 14 15 16
vadd=9 10 11 12 13 14 15 16
msk_neq=0 0 0 0 0 0 0 0
vones=-1 -1 -1 -1 -1 -1 -1 -1
vout=9 10 11 12 13 14 15 16