已合并
[Bugfix] Use RINT cast in experimental sqrt #2848
Robell创建于 5月20日
[Bugfix] Use RINT cast in experimental sqrt #2848
已合并
共 1 个文件变更+1-1
| @@ -112,7 +112,7 @@ __aicore__ inline void KernelSqrt<TYPE_X, TYPE_Y>::Compute(int32_t progress) | |||
| 112 | AscendC::PipeBarrier<PIPE_V>(); | 112 | AscendC::PipeBarrier<PIPE_V>(); |
| 113 | AscendC::Sqrt(p1, p1, this->processDataNum); | 113 | AscendC::Sqrt(p1, p1, this->processDataNum); |
| 114 | AscendC::PipeBarrier<PIPE_V>(); | 114 | AscendC::PipeBarrier<PIPE_V>(); |
| 115 | - AscendC::Cast(yLocal, p1, AscendC::RoundMode::CAST_CEIL, this->processDataNum); | 115 | + AscendC::Cast(yLocal, p1, AscendC::RoundMode::CAST_RINT, this->processDataNum); |
| 116 | } else { | 116 | } else { |
| 117 | AscendC::Sqrt(yLocal, xLocal, this->processDataNum); | 117 | AscendC::Sqrt(yLocal, xLocal, this->processDataNum); |
| 118 | } | 118 | } |