From 7b8010860a1281d9ff5871516d6293e5a841d0ea Mon Sep 17 00:00:00 2001 From: gongchensu Date: Tue, 24 Mar 2026 14:09:41 +0800 Subject: [PATCH] feat(ops): add Moore backend for RmsNorm. --- src/moore/rms_norm/kernel.h | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 src/moore/rms_norm/kernel.h diff --git a/src/moore/rms_norm/kernel.h b/src/moore/rms_norm/kernel.h new file mode 100644 index 00000000..01441e57 --- /dev/null +++ b/src/moore/rms_norm/kernel.h @@ -0,0 +1,25 @@ +#ifndef INFINI_OPS_MOORE_RMS_NORM_KERNEL_H_ +#define INFINI_OPS_MOORE_RMS_NORM_KERNEL_H_ + +#include + +// clang-format off +#include +// clang-format on + +#include "cuda/rms_norm/kernel.h" +#include "moore/caster.cuh" +#include "moore/runtime_.h" + +namespace infini::ops { + +template <> +class Operator + : public CudaRmsNorm> { + public: + using CudaRmsNorm>::CudaRmsNorm; +}; + +} // namespace infini::ops + +#endif