Enabling 64bit to have good accuracy

BQSKit · Jul 1, 2024 · 076bc7b · 076bc7b
1 parent cef123a
commit 076bc7b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_single_leg_tensor.py b/tests/test_single_leg_tensor.py
@@ -3,6 +3,7 @@
 from random import randint
 from random import sample
 
+import jax
 import jax.numpy as jnp
 import pytest
 from bqskit.ir.gates import CXGate
@@ -15,6 +16,9 @@
 from qfactorjax.unitarymatrixjax import UnitaryMatrixJax
 
 
+jax.config.update('jax_enable_x64', True)
+
+
 @pytest.mark.parametrize(
     'num_qubits, N',
     [(randint(2, 7), randint(3, 10)) for _ in range(6)],