Skip to content

Commit 9a02663

Browse files
authored
Add files via upload
1 parent 763ec0e commit 9a02663

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

multioptpy/Calculator/ase_calculation_tools.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -380,12 +380,25 @@ def setup_calculator(atom_obj, software_type, electric_charge_and_multiplicity,
380380
software_path=software_path_dict.get("uma-s-1"),
381381
software_type=software_type)
382382
return calc_obj
383+
384+
if software_type == "uma-s-1p1-cuda":
385+
calc_obj = ASE_FAIRCHEM(atom_obj=atom_obj,
386+
electric_charge_and_multiplicity=electric_charge_and_multiplicity,
387+
software_path=software_path_dict.get("uma-s-1p1"),
388+
software_type=software_type,
389+
device_mode="cuda")
390+
return calc_obj
391+
383392
if software_type == "uma-s-1p1":
384393
calc_obj = ASE_FAIRCHEM(atom_obj=atom_obj,
385394
electric_charge_and_multiplicity=electric_charge_and_multiplicity,
386395
software_path=software_path_dict.get("uma-s-1p1"),
387396
software_type=software_type)
388397
return calc_obj
398+
399+
400+
401+
389402
if software_type == "uma-m-1p1":
390403
calc_obj = ASE_FAIRCHEM(atom_obj=atom_obj,
391404
electric_charge_and_multiplicity=electric_charge_and_multiplicity,

0 commit comments

Comments
 (0)