add feature test

Signed-off-by: Juncheng Gu <jcgu@google.com>

Author: juncgu-google

.buildkite/features/KV_Cache_Offload.yml

@@ -0,0 +1,51 @@
+# KV Cache Offload
+# feature support matrix
+steps:
+  - label: "Correctness tests for KV Cache Offload"
+    key: "KV_Cache_Offload_CorrectnessTest"
+    soft_fail: true
+    agents:
+      queue: tpu_v6e_queue
+    commands:
+      - |
+        .buildkite/scripts/run_in_docker.sh \
+          python3 -m pytest -s -v /workspace/tpu_inference/tests/distributed/offload/tpu_offload_connector_scheduler_test.py \
+          /workspace/tpu_inference/tests/distributed/offload/tpu_offload_connector_worker_test.py \
+          /workspace/tpu_inference/tests/distributed/offload/tpu_offload_cpu_backend_test.py \
+          /workspace/tpu_inference/tests/distributed/offload/tpu_offload_manager_test.py \
+          /workspace/tpu_inference/tests/distributed/offload/tpu_offload_utils_test.py \
+          /workspace/tpu_inference/tests/distributed/offload/tpu_offload_accuracy_test.py
+  - label: "Record correctness test result for KV Cache Offload"
+    key: "record_KV_Cache_Offload_CorrectnessTest"
+    depends_on: "KV_Cache_Offload_CorrectnessTest"
+    env:
+      CI_TARGET: "KV Cache Offload"
+      CI_STAGE: "CorrectnessTest"
+      CI_CATEGORY: "feature support matrix"
+    agents:
+      queue: cpu
+    commands:
+      - |
+        .buildkite/scripts/record_step_result.sh KV_Cache_Offload_CorrectnessTest
+
+  - label: "Performance tests for KV Cache Offload"
+    key: "KV_Cache_Offload_PerformanceTest"
+    depends_on: "record_KV_Cache_Offload_CorrectnessTest"
+    soft_fail: true
+    agents:
+      queue: tpu_v6e_queue
+    commands:
+      - |
+        buildkite-agent meta-data set "KV_Cache_Offload_PerformanceTest" "to be added"
+  - label: "Record performance test result for KV Cache Offload"
+    key: "record_KV_Cache_Offload_PerformanceTest"
+    depends_on: "KV_Cache_Offload_PerformanceTest"
+    env:
+      CI_TARGET: "KV Cache Offload"
+      CI_STAGE: "PerformanceTest"
+      CI_CATEGORY: "feature support matrix"
+    agents:
+      queue: cpu
+    commands:
+      - |
+        .buildkite/scripts/record_step_result.sh KV_Cache_Offload_PerformanceTest