From 68dfa7adcc15eca5f4934d1db21c1274e461dabe Mon Sep 17 00:00:00 2001
From: daysgone <xymdaysgone@qq.com>
Date: Sat, 18 Oct 2025 11:10:45 +0800
Subject: [PATCH] fix import missing

---
 S1/9/rmsnorm_cuda.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/S1/9/rmsnorm_cuda.py b/S1/9/rmsnorm_cuda.py
index 904f3cc..8a0face 100644
--- a/S1/9/rmsnorm_cuda.py
+++ b/S1/9/rmsnorm_cuda.py
@@ -1,4 +1,5 @@
 import torch
+import torch.nn as nn
 from torch.utils.cpp_extension import load_inline
 
 rmsnorm_source = """