From bb0a221e4de726fd84b5e648f775685b8fd58d2c Mon Sep 17 00:00:00 2001
From: Chunyu <15750543867@163.com>
Date: Sat, 8 Nov 2025 11:23:34 +0800
Subject: [PATCH] add rmsnorm kernels support for npu

---
 src/transformers/integrations/hub_kernels.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/transformers/integrations/hub_kernels.py b/src/transformers/integrations/hub_kernels.py
index 112ac670e9a0..673780654872 100644
--- a/src/transformers/integrations/hub_kernels.py
+++ b/src/transformers/integrations/hub_kernels.py
@@ -71,6 +71,12 @@
                     layer_name="RMSNorm",
                 )
             },
+            "npu": {
+                Mode.INFERENCE: LayerRepository(
+                    repo_id="kernels-community/liger_kernels",
+                    layer_name="LigerRMSNorm",
+                )
+            },
         },
         "MLP": {
             "cuda": LayerRepository(