From bea6b1c16659305fe80538017343a5c23f550828 Mon Sep 17 00:00:00 2001 From: Liwei Yang Date: Wed, 19 Apr 2023 15:09:01 +0800 Subject: [PATCH] Remove the test in __main__.py because the test has been moved into folder --- src/qibotn/__main__.py | 21 +-------------------- 1 file changed, 1 insertion(+), 20 deletions(-) diff --git a/src/qibotn/__main__.py b/src/qibotn/__main__.py index f79a24e..15a775a 100644 --- a/src/qibotn/__main__.py +++ b/src/qibotn/__main__.py @@ -57,24 +57,5 @@ def run_bench(task, label): return result -def main_cuquantum(args: argparse.Namespace): - print("Testing for %d nqubits" % (args.nqubits)) - datatype = args.precision - - if args.circuit.lower() == "qft": - circuit = QFT(args.nqubits) - else: - raise NotImplementedError(f"Cannot find circuit {args.circuit}.") - - myconvertor = QiboCircuitToEinsum(circuit, dtype=datatype) - operands_expression = myconvertor.state_vector() - - result_qibo = run_bench(circuit, "Qibo") - sv_cutn = run_bench(lambda: contract(*operands_expression), "cuQuantum cuTensorNet") - - # print(f"is sv in agreement?", cp.allclose(sv_cutn.flatten(), result_qibo.state(numpy=True))) - assert cp.allclose(sv_cutn.flatten(), result_qibo.state(numpy=True)) - - if __name__ == "__main__": - main_cuquantum(parser_cuquantum()) + main(parser())