Open OlfwayAdbayIgbay opened 2 weeks ago
Hi @OlfwayAdbayIgbay ,
Can you confirm which Allegro pair_style
you are using in your LAMMPS script? And which branch of pair_allegro
you are building, and whether you pulled its latest version?
Thanks.
Hi @Linux-cpp-lisp, Thanks for the response. I pulled the most recent version of the pair_allegro main branch. The pair_style looks like this:
pair_style allegro
pair_coeff * * [my path]/deployed_model_latest_run.pth C H N O S
mass 1 1.0
mass 2 1.0
mass 3 1.0
mass 4 1.0
mass 5 1.0
Before that, if of interest, I am initializing like this:
units metal
atom_style atomic
newton on
thermo 1
# get a box defined before pair_coeff
boundary s s s
read_data structure.data
I trained an allegro model using nequip-train and compiled it using nequip-deploy The model was trained with the dtypes:
I am training/infering on a RTX 4090.
LAMMPS is compiled with Kokkos support, patched with the patch_lammps.sh file from pair allegro.
Everything works fine, until I hit the following error:
I have tried casting the variables in question in allegro.py and fc.py to double and float where needed, but eventually I hit a wall where the error I get turns to: