[bugfix] mark eps as dns arg instead of constexpr (#295)

Co-authored-by: unknown <[email protected]>
FlagOpen · Nov 18, 2024 · f27d6b0 · f27d6b0
1 parent 28bcb60
commit f27d6b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/flag_gems/ops/weightnorm.py b/src/flag_gems/ops/weightnorm.py
@@ -236,7 +236,7 @@ def norm_kernel(
     v_shape0,
     v_shape1,
     v_shape2,
-    eps: tl.constexpr,
+    eps,
     BLOCK_ROW_SIZE: tl.constexpr,
     BLOCK_COL_SIZE: tl.constexpr,
 ):
@@ -274,7 +274,7 @@ def norm_bwd_kernel(
     v_shape0,
     v_shape1,
     v_shape2,
-    eps: tl.constexpr,
+    eps,
     BLOCK_ROW_SIZE: tl.constexpr,
     BLOCK_COL_SIZE: tl.constexpr,
 ):