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

FlagOpen · Oct 16, 2024 · 8f669d7 · 8f669d7
1 parent 4e50812
commit 8f669d7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/flag_gems/ops/weightnorm.py b/src/flag_gems/ops/weightnorm.py
@@ -44,7 +44,7 @@ def weight_norm_kernel_last(
     g,
     M,
     N,
-    eps: tl.constexpr,
+    eps,
     BLOCK_ROW_SIZE: tl.constexpr,
     BLOCK_COL_SIZE: tl.constexpr,
 ):
@@ -84,7 +84,7 @@ def weight_norm_kernel_first(
     g,
     M,
     N,
-    eps: tl.constexpr,
+    eps,
     BLOCK_ROW_SIZE: tl.constexpr,
     BLOCK_COL_SIZE: tl.constexpr,
 ):
@@ -126,7 +126,7 @@ def weight_norm_bwd_kernel_last(
     norm,
     M,
     N,
-    eps: tl.constexpr,
+    eps,
     BLOCK_ROW_SIZE: tl.constexpr,
     BLOCK_COL_SIZE: tl.constexpr,
 ):
@@ -176,7 +176,7 @@ def weight_norm_bwd_kernel_first(
     norm,
     M,
     N,
-    eps: tl.constexpr,
+    eps,
     BLOCK_ROW_SIZE: tl.constexpr,
     BLOCK_COL_SIZE: tl.constexpr,
 ):