|
from peachpy import * |
|
from peachpy.x86_64 import * |
|
|
|
|
|
def fp16_alt_xmm_to_fp32_ymm(xmm_half): |
|
ymm_half = YMMRegister() |
|
VPERMQ(ymm_half, xmm_half.as_ymm, 0b01010000) |
|
|
|
ymm_zero = YMMRegister() |
|
VPXOR(ymm_zero.as_xmm, ymm_zero.as_xmm, ymm_zero.as_xmm) |
|
|
|
ymm_word = YMMRegister() |
|
VPUNPCKLWD(ymm_word, ymm_zero, ymm_half) |
|
|
|
ymm_shl1_half = YMMRegister() |
|
VPADDW(ymm_shl1_half, ymm_half, ymm_half) |
|
|
|
ymm_shl1_nonsign = YMMRegister() |
|
VPADDD(ymm_shl1_nonsign, ymm_word, ymm_word) |
|
|
|
sign_mask = Constant.float32x8(-0.0) |
|
|
|
ymm_sign = YMMRegister() |
|
VANDPS(ymm_sign, ymm_word, sign_mask) |
|
|
|
ymm_shr3_nonsign = YMMRegister() |
|
VPSRLD(ymm_shr3_nonsign, ymm_shl1_nonsign, 4) |
|
|
|
exp_offset = Constant.uint32x8(0x38000000) |
|
|
|
ymm_norm_nonsign = YMMRegister() |
|
VPADDD(ymm_norm_nonsign, ymm_shr3_nonsign, exp_offset) |
|
|
|
magic_mask = Constant.uint16x16(0x3E80) |
|
ymm_denorm_nonsign = YMMRegister() |
|
VPUNPCKLWD(ymm_denorm_nonsign, ymm_shl1_half, magic_mask) |
|
|
|
magic_bias = Constant.float32x8(0.25) |
|
VSUBPS(ymm_denorm_nonsign, ymm_denorm_nonsign, magic_bias) |
|
|
|
ymm_denorm_cutoff = YMMRegister() |
|
VMOVDQA(ymm_denorm_cutoff, Constant.uint32x8(0x00800000)) |
|
|
|
ymm_denorm_mask = YMMRegister() |
|
VPCMPGTD(ymm_denorm_mask, ymm_denorm_cutoff, ymm_shr3_nonsign) |
|
|
|
ymm_nonsign = YMMRegister() |
|
VBLENDVPS(ymm_nonsign, ymm_norm_nonsign, ymm_denorm_nonsign, ymm_denorm_mask) |
|
|
|
ymm_float = YMMRegister() |
|
VORPS(ymm_float, ymm_nonsign, ymm_sign) |
|
|
|
return ymm_float |
|
|