Skip to content

Commit b19d8f0

Browse files
committed
chore: remove redundant parameter asserts in RecurrentStatePool
1 parent 1a8792d commit b19d8f0

1 file changed

Lines changed: 0 additions & 10 deletions

File tree

python/sgl_jax/srt/mem_cache/recurrent_state_pool.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -83,16 +83,6 @@ def __init__(
8383
# total_slots: size+1 (for dummy slot 0), ceil to dp_size
8484
self.total_slots = _ceil_to(size + 1, dp_size)
8585

86-
assert size > 0, f"size must be > 0, got {size}"
87-
assert num_heads > 0
88-
assert head_dim > 0
89-
assert num_k_heads > 0
90-
assert head_k_dim > 0
91-
assert (
92-
conv_kernel_size >= 2
93-
), f"conv_kernel_size must be >= 2 (got {conv_kernel_size}); K=1 produces empty conv buffers."
94-
assert self.proj_size > 0
95-
9686
self.mesh = mesh
9787
self.recurrent_partition_axis = recurrent_partition_axis
9888
self.conv_partition_axis = conv_partition_axis

0 commit comments

Comments
 (0)