Add occupancy hint to rms_norm

haijieg · haijieg · commit 3ce4742a9ecd · 2026-01-19T12:00:29.000-08:00
Signed-off-by: Jay Gu &lt;jagu@nvidia.com&gt;
diff --git a/test/kernels/rms_norm.py b/test/kernels/rms_norm.py
@@ -6,7 +6,7 @@
 import numpy as np
 
 
-@ct.kernel
+@ct.kernel(occupancy=ct.ByTarget(sm_100=16))
 def rms_norm_kernel(
     x,
     w,
