From a5d18b7dd03219056ded0a7c6ee8daf673a88cc2 Mon Sep 17 00:00:00 2001 From: positr0nium Date: Mon, 17 Jun 2024 10:12:45 +0200 Subject: [PATCH] reactivated ML based sparse matrix mult algorithm decision --- src/qrisp/simulator/bi_array_helper.py | 2 +- tests/test_unitary_calculation.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/qrisp/simulator/bi_array_helper.py b/src/qrisp/simulator/bi_array_helper.py index eacc84a2..d02225b4 100644 --- a/src/qrisp/simulator/bi_array_helper.py +++ b/src/qrisp/simulator/bi_array_helper.py @@ -541,7 +541,7 @@ def sparse_matrix_mult(A, B): log_sparsity_a = -np.log2(B.nnz/(B.shape[0]*B.shape[1])) log_sparsity_b = -np.log2(A.nnz/(A.shape[0]*A.shape[1])) - if get_prediction(log_shape_0_a, log_shape_1_b, log_sparsity_a, log_sparsity_b) and False: + if get_prediction(log_shape_0_a, log_shape_1_b, log_sparsity_a, log_sparsity_b): return (A @ B).tocoo() else: return coo_sparse_matrix_mult(A, B) diff --git a/tests/test_unitary_calculation.py b/tests/test_unitary_calculation.py index b3f92e98..d8b6e921 100644 --- a/tests/test_unitary_calculation.py +++ b/tests/test_unitary_calculation.py @@ -36,9 +36,9 @@ def test_unitary_calculation(): x = QuantumFloat(n, 0, qs, signed=True) y = QuantumFloat(n, 0, qs, signed=False) - s = x * y + s = x + y # qs = s.qs - qc = qs.compile(1) + qc = qs.compile() start = time.time() test_unitary_1 = qc.get_unitary()