Skip to content

Commit 6dc060b

Browse files
fix(gpu): enable gemm
1 parent 26b9676 commit 6dc060b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/tfhe-cuda-backend/cuda/src/crypto/keyswitch.cuh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -404,7 +404,7 @@ void execute_keyswitch_async(CudaStreams streams,
404404
Torus *current_lwe_input_indexes =
405405
get_variant_element(lwe_input_indexes, i);
406406

407-
if (false && (num_samples_on_gpu >= 144)) {
407+
if (num_samples_on_gpu >= 144) {
408408
// Compute Keyswitch
409409
host_gemm_keyswitch_lwe_ciphertext_vector<Torus>(
410410
streams.stream(i), streams.gpu_index(i), current_lwe_array_out,

0 commit comments

Comments
 (0)