xref: /aosp_15_r20/external/executorch/codegen/templates/RegisterKernels.cpp (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree.
 */

// ${generated_comment}
// This implements register_all_kernels() API that is declared in
// RegisterKernels.h
#include "RegisterKernels.h"
#include "${fn_header}" // Generated Function import headers

namespace torch {
namespace executor {

Error register_all_kernels() {
  Kernel kernels_to_register[] = {
      ${unboxed_kernels} // Generated kernels
  };
  Error success_with_kernel_reg =
      ::executorch::runtime::register_kernels({kernels_to_register});
  if (success_with_kernel_reg != Error::Ok) {
    ET_LOG(Error, "Failed register all kernels");
    return success_with_kernel_reg;
  }
  return Error::Ok;
}

} // namespace executor
} // namespace torch