Skip to content

Commit e9f21fb

Browse files
committed
feat(coprocessor): limit GPU CI tests to FHEUint64 type matrix
1 parent 410c7e9 commit e9f21fb

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

.github/workflows/coprocessor-gpu-tests.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -215,8 +215,15 @@ jobs:
215215
echo "Database migration completed"
216216
217217
- name: Run GPU tests for the worker services.
218+
env:
219+
IS_MERGE_QUEUE: ${{ startsWith(github.head_ref, 'mergify/merge-queue/') && '1' || '0' }}
218220
run: |
219221
export DATABASE_URL=postgresql://postgres:postgres@localhost:5432/coprocessor
222+
# Merge queue: leave unset so supported_types() defaults to full matrix.
223+
# PR CI: run only FHEUint64 for faster feedback.
224+
if [ "$IS_MERGE_QUEUE" != "1" ]; then
225+
export TFHE_WORKER_EVENT_TYPE_MATRIX=uint64
226+
fi
220227
cargo test \
221228
-p tfhe-worker \
222229
-p sns-worker \

coprocessor/fhevm-engine/tfhe-worker/src/tests/operators_from_events.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,14 @@ const FULL_SUPPORTED_TYPES: &[i32] = &[
3939
11, // 2048 bit
4040
];
4141

42+
const UINT64_ONLY: &[i32] = &[
43+
5, // 64 bit
44+
];
45+
4246
pub fn supported_types() -> &'static [i32] {
4347
match std::env::var("TFHE_WORKER_EVENT_TYPE_MATRIX") {
4448
Ok(mode) if mode.eq_ignore_ascii_case("local") => LOCAL_SUPPORTED_TYPES,
49+
Ok(mode) if mode.eq_ignore_ascii_case("uint64") => UINT64_ONLY,
4550
_ => FULL_SUPPORTED_TYPES,
4651
}
4752
}

0 commit comments

Comments
 (0)