diff options
Diffstat (limited to 'pseudocode/operators/CAST.tosac')
-rw-r--r-- | pseudocode/operators/CAST.tosac | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pseudocode/operators/CAST.tosac b/pseudocode/operators/CAST.tosac index 8d816ca..fac73e3 100644 --- a/pseudocode/operators/CAST.tosac +++ b/pseudocode/operators/CAST.tosac @@ -17,11 +17,11 @@ for_each(index in shape) { } else if (out_t == fp16_t || out_t == bf16_t || out_t == fp32_t) { out = round_to_nearest_float(in); } else if (in_t == fp16_t || in_t == bf16_t || in_t == fp32_t) { - out = apply_clip<out_t>(round_to_nearest_int(in), minimum<out_t>, maximum<out_t>); + out = truncate<out_t>(apply_clip_s<i32_t>(round_to_nearest_int(in), minimum<out_t>, maximum<out_t>)); } else if (sizeof(out_t) >= sizeof(in_t)) { out = sign_extend<out_t>(in); } else { - out = truncate(in); + out = truncate<out_t>(in); } tensor_write<out_t>(output, shape, index, out); } |