1*4bdc9457SAndroid Build Coastguard Worker // Copyright 2021 Google LLC 2*4bdc9457SAndroid Build Coastguard Worker // 3*4bdc9457SAndroid Build Coastguard Worker // This source code is licensed under the BSD-style license found in the 4*4bdc9457SAndroid Build Coastguard Worker // LICENSE file in the root directory of this source tree. 5*4bdc9457SAndroid Build Coastguard Worker 6*4bdc9457SAndroid Build Coastguard Worker #include <assert.h> 7*4bdc9457SAndroid Build Coastguard Worker #include <stddef.h> 8*4bdc9457SAndroid Build Coastguard Worker #include <stdint.h> 9*4bdc9457SAndroid Build Coastguard Worker 10*4bdc9457SAndroid Build Coastguard Worker #include <arm_neon.h> 11*4bdc9457SAndroid Build Coastguard Worker 12*4bdc9457SAndroid Build Coastguard Worker #include <xnnpack/math-stubs.h> 13*4bdc9457SAndroid Build Coastguard Worker 14*4bdc9457SAndroid Build Coastguard Worker xnn_math_f32_f16_cvt__neonfp16(size_t n,const float * input,void * output)15*4bdc9457SAndroid Build Coastguard Workervoid xnn_math_f32_f16_cvt__neonfp16( 16*4bdc9457SAndroid Build Coastguard Worker size_t n, 17*4bdc9457SAndroid Build Coastguard Worker const float* input, 18*4bdc9457SAndroid Build Coastguard Worker void* output) 19*4bdc9457SAndroid Build Coastguard Worker { 20*4bdc9457SAndroid Build Coastguard Worker assert(n % (4 * sizeof(uint16_t)) == 0); 21*4bdc9457SAndroid Build Coastguard Worker 22*4bdc9457SAndroid Build Coastguard Worker uint16_t* o = (uint16_t*) output; 23*4bdc9457SAndroid Build Coastguard Worker for (; n != 0; n -= 4 * sizeof(uint16_t)) { 24*4bdc9457SAndroid Build Coastguard Worker const float32x4_t vx = vld1q_f32(input); input += 4; 25*4bdc9457SAndroid Build Coastguard Worker const uint16x4_t vy = vreinterpret_u16_f16(vcvt_f16_f32(vx)); 26*4bdc9457SAndroid Build Coastguard Worker vst1_u16(o, vy); o += 4; 27*4bdc9457SAndroid Build Coastguard Worker } 28*4bdc9457SAndroid Build Coastguard Worker } 29