github/tfhe-rs
1
1
Fork 0
mirror of https://github.com/zama-ai/tfhe-rs.git synced 2026-01-08 06:13:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore(gpu): remove m3_c3 & gf 3 params from multi-gpu tests to reduce ci time

Browse Source
This commit is contained in:
Agnes Leroy
2025-06-27 17:56:45 +02:00
committed by Agnès Leroy
parent 187159d9f9
commit 4bba35e926
3 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/gpu_full_multi_gpu_tests.yml vendored
View File

@@ -137,7 +137,7 @@ jobs:
# No need to test core_crypto and classic PBS in integer since it's already tested on single GPU.
- name: Run multi-bit CUDA integer tests
run: |
BIG_TESTS_INSTANCE=TRUE make test_integer_multi_bit_gpu_ci
BIG_TESTS_INSTANCE=TRUE NO_BIG_PARAMS_GPU=TRUE make test_integer_multi_bit_gpu_ci
- name: Run user docs tests
run: |

7
scripts/integer-tests.sh
View File

@@ -27,6 +27,7 @@ fast_tests_argument=
long_tests_argument=
nightly_tests_argument=
no_big_params_argument=
no_big_params_gpu_argument=
cargo_profile="release"
backend="cpu"
gpu_feature=""
@@ -107,6 +108,10 @@ if [[ "${NO_BIG_PARAMS}" == TRUE ]]; then
no_big_params_argument=--no-big-params
fi
if [[ "${NO_BIG_PARAMS_GPU}" == TRUE ]]; then
no_big_params_argument_gpu=--no-big-params-gpu
fi
if [[ "${backend}" == "gpu" ]]; then
gpu_feature="gpu"
fi
@@ -145,7 +150,7 @@ if [[ "${backend}" == "gpu" ]]; then
fi
fi
filter_expression=$(/usr/bin/python3 scripts/test_filtering.py --layer integer --backend "${backend}" ${fast_tests_argument} ${long_tests_argument} ${nightly_tests_argument} ${multi_bit_argument} ${sign_argument} ${no_big_params_argument})
filter_expression=$(/usr/bin/python3 scripts/test_filtering.py --layer integer --backend "${backend}" ${fast_tests_argument:+$fast_tests_argument} ${long_tests_argument:+$long_tests_argument} ${nightly_tests_argument:+$nightly_tests_argument} ${no_big_params_argument_gpu:+$no_big_params_argument_gpu} ${multi_bit_argument:+$multi_bit_argument} ${sign_argument:+$sign_argument} ${no_big_params_argument:+$no_big_params_argument})
if [[ "${FAST_TESTS}" == "TRUE" ]]; then
echo "Running 'fast' test set"

14
scripts/test_filtering.py
View File

@@ -68,6 +68,12 @@ parser.add_argument(
action="store_true",
help="Do not run tests with big parameters set (e.g. 4bits message with 4 bits carry)",
)
parser.add_argument(
"--no-big-params-gpu",
dest="no_big_params_gpu",
action="store_true",
help="Do not run tests with big parameters set (e.g. 3bits message with 3 bits carry) for GPU",
)
# block PBS are too slow for high params
# mul_crt_4_4 is extremely flaky (~80% failure)
@@ -101,6 +107,10 @@ EXCLUDED_BIG_PARAMETERS = [
"/.*_param_message_4_carry_4_ks_pbs_gaussian_2m64$/",
]
EXCLUDED_BIG_PARAMETERS_GPU = [
"/.*_message_3_carry_3.*$/",
"/.*_group_3_message_2_carry_2.*$/",
]
def filter_integer_tests(input_args):
(multi_bit_filter, group_filter) = (
@@ -130,6 +140,10 @@ def filter_integer_tests(input_args):
for pattern in EXCLUDED_BIG_PARAMETERS:
filter_expression.append(f"not test({pattern})")
if input_args.no_big_params_gpu:
for pattern in EXCLUDED_BIG_PARAMETERS_GPU:
filter_expression.append(f"not test({pattern})")
if input_args.fast_tests and input_args.nightly_tests:
filter_expression.append(
f"test(/.*_default_.*?_param{multi_bit_filter}{group_filter}_message_[2-3]_carry_[2-3]_.*/)"