Line data    Source code

       1            0 : // Distributed under the MIT License.
       2              : // See LICENSE.txt for details.
       3              : 
       4              : #pragma once
       5              : 
       6              : #ifdef SPECTRE_AUTODIFF
       7              : 
       8              : #include <autodiff/common/numbertraits.hpp>
       9              : #include <autodiff/forward/dual.hpp>
      10              : #include <autodiff/reverse/var.hpp>
      11              : 
      12              : #include "Utilities/MakeWithValue.hpp"
      13              : #include "Utilities/Simd/Simd.hpp"
      14              : 
      15            0 : namespace autodiff {
      16            0 : using BatchType = simd::batch<double>;
      17            0 : using SecondOrderDual = autodiff::HigherOrderDual<2, BatchType>;
      18            0 : using SecondOrderDualNum = autodiff::HigherOrderDual<2, double>;
      19              : 
      20              : namespace detail {
      21              : /// Template specialization for simd::batch<double> to treat it as arithmetic.
      22              : // The major difficulty we have with DataVector working with autodiff is
      23              : // DataVector does not have a scalar broadcast constructor, which is expected
      24              : // in the seed function for autodiff dual type.
      25              : template <>
      26              : struct ArithmeticTraits<simd::batch<double>> {
      27              :   static constexpr bool isArithmetic = true;
      28              : };
      29              : }  // namespace detail
      30              : }  // namespace autodiff
      31              : 
      32              : namespace MakeWithValueImpls {
      33              : template <typename T>
      34            0 : struct MakeWithValueImpl<autodiff::HigherOrderDual<2, double>, T> {
      35            0 :   static SPECTRE_ALWAYS_INLINE autodiff::HigherOrderDual<2, double> apply(
      36              :       const T& /* input */, const double value) {
      37              :     return {value};
      38              :   }
      39              : };
      40              : 
      41              : template <typename T>
      42            0 : struct MakeWithValueImpl<autodiff::HigherOrderDual<2, simd::batch<double>>, T> {
      43              :   static SPECTRE_ALWAYS_INLINE autodiff::HigherOrderDual<2, simd::batch<double>>
      44            0 :   apply(const T& /* input */, const double value) {
      45              :     return {simd::batch<double>::broadcast(value)};
      46              :   }
      47              : };
      48              : }  // namespace MakeWithValueImpls
      49              : 
      50              : #endif  // SPECTRE_AUTODIFF