update

MzeroMiko · Feb 25, 2024 · 7ef968b · 7ef968b
1 parent 6529bea
commit 7ef968b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,3 +2,4 @@
 __pycache__
 test_selective_scan_test.py
 .test_selective_scan.py
+1.log
diff --git a/test_selective_scan.py b/test_selective_scan.py
@@ -480,6 +480,7 @@ def selective_scan_ref(u, delta, A, B, C, D=None, z=None, delta_bias=None, delta
 @pytest.mark.parametrize("is_variable_C", [True])
 @pytest.mark.parametrize("is_variable_B", [True])
 @pytest.mark.parametrize("chunksize", [64])
+# @pytest.mark.parametrize("chunksize", [32])
 def test_selective_scan(is_variable_B, is_variable_C, varBC_groups, has_D, has_z, has_delta_bias,
                         delta_softplus, return_last_state, seqlen, itype, wtype, chunksize):
     selective_scan_fn = build_api_selective_scan(chunksize=chunksize)