From 8d4d0468bdb080be398af8b8aaf22aa323522ec0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Joseph=20Hopfm=C3=BCller?= Date: Wed, 20 Nov 2024 22:56:26 +0100 Subject: [PATCH] complexhalf (complex32) isn't supported by torch.linalg.qr --- src/single-core-regen/regen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/single-core-regen/regen.py b/src/single-core-regen/regen.py index 7dbdd3e..6f463e5 100644 --- a/src/single-core-regen/regen.py +++ b/src/single-core-regen/regen.py @@ -15,7 +15,7 @@ global_settings = GlobalSettings( data_settings = DataSettings( config_path = "data/*-128-16384-100000-0-0-17-0-PAM4-0.ini", - dtype = ("complex64", "float64", "complex32", "float32"), + dtype = ("complex128", "complex64", "float64", "float32"), symbols = (1, 16), model_input_dim = (1, 32), shuffle = True,