Repository navigation
Release 0.1.2: early GRACE-1L/2L models, padded GPU kernels for more fits - #13
Merged
Merged
Conversation
…fits - The early GRACE-1L-OAM, GRACE-1L-OMAT, GRACE-2L-OAM and GRACE-2L-OMAT models are in the registry and run on the CPU, NVIDIA and AMD GPUs, in Python and in LAMMPS. Own fits of the early presets export as kokkos.npz format 2. - GPUs: GRACE-1L/2L fits narrower than compiled kernels run zero-padded on them, with a warning. Wider GRACE-2L fits stop with an error, or run on generic kernels with GRACE_GENERIC_KERNELS=1. - device="auto" raises for such a wider GRACE-2L fit instead of running on the CPU. - GPU module interface 7; older modules are refused. - Tests: padded models against the originals, legacy models against TensorFlow fp64 references. See CHANGELOG.md.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Release 0.1.2: the files of the tag 0.1.2 (one commit). Merge with Rebase and merge.
The early GRACE-1L/2L models, and fast GPU kernels for more GRACE-1L/2L fits.
Added
are in the registry and run on the CPU, NVIDIA and AMD GPUs, in Python and in LAMMPS, without an uncertainty head.
Own fits of the early GRACE-1L/2L presets run as well: tensorpotential exports them as
kokkos.npzformat 2.zero-padded to them, with the same results and a warning that names the widths (not models with an uncertainty
head). GRACE-2L has kernels for the widths of the current registry models, of the early ones, and of fits with 48
layer-2 radial channels; GRACE-1L fits run with up to 64 radial channels. The models guide lists the widths.
GRACE_GENERIC_KERNELS=1: a GRACE-2L fit wider than all compiled GPU kernels runs on generic GPU kernels, severaltimes slower, with a warning. Without the variable such a fit stops with an error that names its widths.
LAMMPS (
grace-runtimebranch of the LAMMPS fork)grace/runtimeandgrace/runtime/kkrun the early GRACE-1L/2L models and own fits of their presets. The Kokkosstyle pads narrower fits like Python does, and with
GRACE_GENERIC_KERNELS=1runs wider GRACE-2L fits on thegeneric GPU kernels. Warnings appear as LAMMPS
WARNINGlines.Documentation
GRACE_GENERIC_KERNELS.Changed defaults
zero-padded with a warning; a wider GRACE-2L one still stops, and the error names
GRACE_GENERIC_KERNELS=1.device="auto": such a wider GRACE-2L model raises instead of running on the CPU.kokkos.npzformat version 2 for the early GRACE-1L/2L layouts. grace-runtime 0.1.1 and older refuse these filesand ask for an update. Files of current models stay format 1.
grace-runtime-cuda12/grace-runtime-rocm60.1.2 have interface version 7; grace-runtime0.1.2 refuses older modules. The packages pin each other, so a normal
pip installmatches them.