Program Listing for File soft_min_switch.h

↰ Return to documentation for file (source/lib/include/soft_min_switch.h)

// SPDX-License-Identifier: LGPL-3.0-or-later
#pragma once

namespace deepmd {

template <typename FPTYPE>
void soft_min_switch_cpu(FPTYPE* sw_value,
                         FPTYPE* sw_deriv,
                         const FPTYPE* rij,
                         const int* nlist,
                         const int& nloc,
                         const int& nnei,
                         const FPTYPE& alpha,
                         const FPTYPE& rmin,
                         const FPTYPE& rmax);

}