aie::set_rounding()
和 aie::set_saturation()
API 用于为累加结果设置舍入和饱和模式。
aie::rounding_mode
包括:-
floor
- 无舍入 - 在 LSB 处截位,始终向下舍入(趋向负无穷)。 -
ceil
- 无舍入 - 始终向上舍入(趋向正无穷)。 -
positive_inf
- 半舍入趋向正无穷。 -
negative_inf
- 半舍入趋向负无穷。 -
symmetric_inf
- 半舍入趋向无穷(远离 0)。 -
symmetric_zero
- 半舍入趋向零(远离无穷)。 -
conv_even
- 半舍入趋向最接近的偶数。 -
conv_odd
- 半舍入趋向最接近的奇数。
aie::saturation_mode
包括:-
none
- 不执行饱和,该值于 MSB 侧执行截位。 -
truncate
- 控制饱和。饱和会对范围 [- ( 2^(n-1) ) : +2^(n-1) - 1
] 内的 n 位有符号值进行舍入。例如,如果 n=8,那么范围为 [-128:127]。 -
symmetric
- 控制对称饱和。对称饱和会对范围 [- ( 2^(n-1) -1 ) : +2^(n-1) - 1
] 内的 n 位有符号值进行舍入。例如,如果 n=8,那么范围为 [-127:127]。
舍入和饱和设置适用于 AI 引擎拼块。aie::get_rounding
和 aie::get_saturation
方法可用于获取当前 AI 引擎拼块的舍入和饱和模式。
以下提供的代码示例演示了如何设置和获取舍入和饱和模式:
aie::set_rounding(aie::rounding_mode::ceil);
aie::set_saturation(aie::saturation_mode::truncate);
aie::rounding_mode current_rnd=aie::get_rounding();
aie::saturation_mode current_sat=aie::get_saturation();
...
aie::set_rounding(aie::rounding_mode::floor);
aie::set_saturation(aie::saturation_mode::none);
注释: 舍入和饱和设置均为粘性设置。这表示 AI 引擎拼块会保留其模式直至更改。如果内核的运行时比率小于 1,则可与其它内核共享拼块,如果舍入和饱和模式影响内核操作,那么内核必须在内核代码中而不是在构造器或初始化函数中设置这些模式。这将确保当相同拼块上的其它内核使用不同的舍入或饱和模式值时,可保证舍入和饱和模式。