sglang_v0.5.2/pytorch_2.8.0/third_party/XNNPACK/test/f16-qu8-vcvt.cc

49 lines
3.0 KiB
C++

// Copyright 2024 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
#include "xnnpack/microparams-init.h"
#include "xnnpack/vcvt.h"
#include "vunary-microkernel-tester.h"
#define XNN_QUANTIZED(T) xnnpack::quantized<T>
#define XNN_CVT_UKERNEL_WITH_PARAMS(arch_flags, ukernel, batch_tile, \
vector_tile, datatype_in, datatype_out, \
params_type, init_params) \
TEST(ukernel, batch_eq) { \
TestBatchEq<Convert, datatype_in, datatype_out>(arch_flags, batch_tile, \
ukernel, init_params); \
} \
TEST(ukernel, batch_div) { \
TestBatchDiv<Convert, datatype_in, datatype_out>(arch_flags, batch_tile, \
ukernel, init_params); \
} \
TEST(ukernel, batch_lt) { \
TestBatchLT<Convert, datatype_in, datatype_out>(arch_flags, batch_tile, \
ukernel, init_params); \
} \
TEST(ukernel, batch_gt) { \
TestBatchGT<Convert, datatype_in, datatype_out>(arch_flags, batch_tile, \
ukernel, init_params); \
} \
TEST(ukernel, output_scale) { \
TestOutputScale<Convert, datatype_in, datatype_out>( \
arch_flags, batch_tile, ukernel, init_params); \
} \
TEST(ukernel, output_zero_point) { \
TestOutputZeroPoint<Convert, datatype_in, datatype_out>( \
arch_flags, batch_tile, ukernel, init_params); \
} \
TEST(ukernel, output_saturation) { \
TestOutputSaturation<Convert, datatype_in, datatype_out>( \
arch_flags, batch_tile, ukernel, init_params); \
} \
TEST(ukernel, output_overflow) { \
TestOutputOverflow<Convert, datatype_in, datatype_out>( \
arch_flags, batch_tile, ukernel, init_params); \
}
#include "f16-qu8-vcvt/f16-qu8-vcvt.h"
#undef XNN_CVT_UKERNEL_WITH_PARAMS
#undef XNN_QUANTIZED