add HIP platform to relax

fastfold/relax/amber_minimize.py

@@ -94,8 +94,8 @@ def _openmm_minimize(
         _add_restraints(system, pdb, stiffness, restraint_set, exclude_residues)
 
     integrator = openmm.LangevinIntegrator(0, 0.01, 0.0)
-    # platform = openmm.Platform.getPlatformByName("CUDA" if use_gpu else "CPU")
-    simulation = openmm_app.Simulation(pdb.topology, system, integrator)
+    platform = openmm.Platform.getPlatformByName("HIP" if use_gpu else "CPU")
+    simulation = openmm_app.Simulation(pdb.topology, system, integrator, platform)
     simulation.context.setPositions(pdb.positions)
 
     ret = {}
@@ -536,6 +536,7 @@ def get_initial_energies(
         openmm_pdbs[0].topology,
         system,
         openmm.LangevinIntegrator(0, 0.01, 0.0),
+        openmm.Platform.getPlatformByName("CPU"),
     )
     energies = []
     for pdb in openmm_pdbs: