xref: /aosp_15_r20/external/pytorch/test/edge/templates/RegisterCodegenUnboxedKernels.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

#include <operator_registry.h>
#include <event_tracer_hooks.h>
#include "${fn_header}" // Generated Function import headers

namespace torch {
namespace executor {

using namespace internal;

namespace {
using KernelArrayRef = ::at::ArrayRef<::torch::executor::Kernel>;

static Kernel kernels_to_register[] = {
    ${unboxed_kernels} // Generated operators
};

// Explicitly convert to ArrayRef, so that the API can take an empty C array of
// Kernels.
static KernelArrayRef kernel_array_ref(
    kernels_to_register,
    kernels_to_register + sizeof(kernels_to_register) / sizeof(Kernel));

// Return value not used. Keep the static variable assignment to register
// operators in static initialization time.
static auto success_with_kernel_reg = register_kernels(kernel_array_ref);
} // namespace
} // namespace executor
} // namespace torch