Set dtype and device in trapzd (#80)

dftd4 · Jan 5, 2025 · a1e1a3b · a1e1a3b
1 parent 2215589
commit a1e1a3b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/tad_dftd4/model.py b/src/tad_dftd4/model.py
@@ -531,7 +531,9 @@ def trapzd(polarizability: Tensor) -> Tensor:
             1.7500000000000000,
             2.5000000000000000,
             1.2500000000000000,
-        ]
+        ],
+        device=polarizability.device,
+        dtype=polarizability.dtype,
     )
 
     # NOTE: In the old version, a memory inefficient intermediate tensor was