fix import missing

2025-10-18 11:10:45 +08:00 · 2025-10-18 11:10:45 +08:00 · 68dfa7adcc
parent 86663cc044
commit 68dfa7adcc
1 changed files with 1 additions and 0 deletions
--- 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 = """