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

25 lines
1.8 KiB
C++

// Copyright 2019 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, input_scale) { TestInputScale<Convert, datatype_in, datatype_out>(arch_flags, batch_tile, ukernel, init_params); } \
TEST(ukernel, input_zero_point) { TestInputZeroPoint<Convert, datatype_in, datatype_out>(arch_flags, batch_tile, ukernel, init_params); }
#include "qu8-vcvt/qu8-vcvt.h"
#undef XNN_CVT_UKERNEL_WITH_PARAMS
#undef XNN_QUANTIZED