We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5c64440 commit 70d3a3cCopy full SHA for 70d3a3c
1 file changed
docs/tutorials/forces_gpu_tutorial.rst
@@ -154,6 +154,7 @@ and use this information however you want.
154
output = np.zeros(1, dtype=sim_specs["out"])
155
output["energy"][0] = final_energy
156
157
+
158
return output
159
160
The above code will assign a GPU to each worker on CUDA-capable systems,
0 commit comments