[debug] make all run same shard_id

Signed-off-by: Yu Chin Fabian Lim <[email protected]>

Author: fabianlim

.buildkite/test-pipeline.yaml

@@ -290,9 +290,11 @@ steps:
   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
-  parallelism: 1
-  #parallelism: 4
+    # - 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
+  parallelism: 4
 
 - label: Tensorizer Test # 11min
   mirror_hardwares: [amd]