Skip to content

Commit 52b148a

Browse files
committed
chore(gpu): temporarily set test threads to 1 for the GPU
1 parent d0624d6 commit 52b148a

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

Makefile

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -473,23 +473,23 @@ test_gpu: test_core_crypto_gpu test_integer_gpu test_cuda_backend
473473
.PHONY: test_core_crypto_gpu # Run the tests of the core_crypto module including experimental on the gpu backend
474474
test_core_crypto_gpu: install_rs_build_toolchain
475475
RUSTFLAGS="$(RUSTFLAGS)" cargo $(CARGO_RS_BUILD_TOOLCHAIN) test --profile $(CARGO_PROFILE) \
476-
--features=$(TARGET_ARCH_FEATURE),gpu -p $(TFHE_SPEC) -- core_crypto::gpu::
476+
--features=$(TARGET_ARCH_FEATURE),gpu -p $(TFHE_SPEC) -- core_crypto::gpu:: --test-threads=1
477477
RUSTFLAGS="$(RUSTFLAGS)" cargo $(CARGO_RS_BUILD_TOOLCHAIN) test --doc --profile $(CARGO_PROFILE) \
478-
--features=$(TARGET_ARCH_FEATURE),gpu -p $(TFHE_SPEC) -- core_crypto::gpu::
478+
--features=$(TARGET_ARCH_FEATURE),gpu -p $(TFHE_SPEC) -- core_crypto::gpu:: --test-threads=1
479479

480480
.PHONY: test_integer_gpu # Run the tests of the integer module including experimental on the gpu backend
481481
test_integer_gpu: install_rs_build_toolchain
482482
RUSTFLAGS="$(RUSTFLAGS)" cargo $(CARGO_RS_BUILD_TOOLCHAIN) test --profile $(CARGO_PROFILE) \
483-
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::server_key:: --test-threads=6
483+
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::server_key:: --test-threads=1
484484
RUSTFLAGS="$(RUSTFLAGS)" cargo $(CARGO_RS_BUILD_TOOLCHAIN) test --doc --profile $(CARGO_PROFILE) \
485485
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::server_key::
486486

487487
.PHONY: test_integer_compression_gpu
488488
test_integer_compression_gpu: install_rs_build_toolchain
489489
RUSTFLAGS="$(RUSTFLAGS)" cargo $(CARGO_RS_BUILD_TOOLCHAIN) test --profile $(CARGO_PROFILE) \
490-
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::ciphertext::compressed_ciphertext_list::tests::
490+
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::ciphertext::compressed_ciphertext_list::tests:: --test-threads=1
491491
RUSTFLAGS="$(RUSTFLAGS)" cargo $(CARGO_RS_BUILD_TOOLCHAIN) test --doc --profile $(CARGO_PROFILE) \
492-
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::ciphertext::compress
492+
--features=$(TARGET_ARCH_FEATURE),integer,gpu -p $(TFHE_SPEC) -- integer::gpu::ciphertext::compress --test-threads=1
493493

494494
.PHONY: test_integer_gpu_ci # Run the tests for integer ci on gpu backend
495495
test_integer_gpu_ci: install_rs_check_toolchain install_cargo_nextest

scripts/integer-tests.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,8 @@ fi
130130
# Override test-threads number to avoid Out-of-memory issues on GPU instances
131131
if [[ "${backend}" == "gpu" ]]; then
132132
if [[ "${BIG_TESTS_INSTANCE}" == TRUE ]]; then
133-
test_threads=8
134-
doctest_threads=8
133+
test_threads=1
134+
doctest_threads=1
135135
else
136136
test_threads=1
137137
doctest_threads=1

0 commit comments

Comments
 (0)