[debug] .contiguous()

Signed-off-by: Yu Chin Fabian Lim <[email protected]>
vllm-project · Feb 4, 2025 · d6d4086 · d6d4086
1 parent 965620d
commit d6d4086
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml
@@ -288,12 +288,7 @@ steps:
   - vllm/attention
   - tests/kernels
   commands:
-    # - pytest -v -s kernels --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
-    # DEBUG
-    # - pytest -v -s kernels/test_mamba_ssm_ssd.py
-    # DEBUG make all run the same shard ID
-    - pytest -v -s kernels --shard-id=2 --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
-  # parallelism: 1
+    - pytest -v -s kernels --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
   parallelism: 4
 
 - label: Tensorizer Test # 11min

diff --git a/tests/kernels/test_mamba_ssm_ssd.py b/tests/kernels/test_mamba_ssm_ssd.py
@@ -275,11 +275,11 @@ def test_mamba_chunk_scan_cont_batch(d_head, n_heads, seq_len_chunk_size_cases,
                                          d_head, itype):
 
         Y, new_states = mamba_chunk_scan_combined(
-            X,
-            dt,
-            A,
-            B,
-            C,
+            X.contiguous(),
+            dt.contiguous(),
+            A.contiguous(),
+            B.contiguous(),
+            C.contiguous(),
             chunk_size,
             D=None,
             cu_seqlens=cu_seqlens,